Kuberwastaken commited on
Commit
48dab84
·
1 Parent(s): 4ffd7e0

Random Improvements

Browse files
Files changed (1) hide show
  1. model/analyzer.py +1 -1
model/analyzer.py CHANGED
@@ -172,7 +172,7 @@ class ContentAnalyzer:
172
  with torch.no_grad():
173
  outputs = self.model.generate(
174
  **inputs,
175
- max_new_tokens=3,
176
  do_sample=True,
177
  temperature=0.7,
178
  top_p=0.8,
 
172
  with torch.no_grad():
173
  outputs = self.model.generate(
174
  **inputs,
175
+ max_new_tokens=20,
176
  do_sample=True,
177
  temperature=0.7,
178
  top_p=0.8,