Spaces:
Running
Running
Kuberwastaken
commited on
Commit
·
63fa8cc
1
Parent(s):
7b4e84e
Made Model work again
Browse files- model/analyzer.py +3 -3
model/analyzer.py
CHANGED
@@ -156,10 +156,10 @@ class ContentAnalyzer:
|
|
156 |
with torch.no_grad():
|
157 |
outputs = self.model.generate(
|
158 |
**inputs,
|
159 |
-
max_new_tokens=
|
160 |
do_sample=True,
|
161 |
-
temperature=0.
|
162 |
-
top_p=0.
|
163 |
pad_token_id=self.tokenizer.eos_token_id
|
164 |
)
|
165 |
|
|
|
156 |
with torch.no_grad():
|
157 |
outputs = self.model.generate(
|
158 |
**inputs,
|
159 |
+
max_new_tokens=3,
|
160 |
do_sample=True,
|
161 |
+
temperature=0.7,
|
162 |
+
top_p=0.8,
|
163 |
pad_token_id=self.tokenizer.eos_token_id
|
164 |
)
|
165 |
|