Spaces:
Running
Running
Commit
·
48dab84
1
Parent(s):
4ffd7e0
Random Improvements
Browse files- 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=
|
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,
|