Kuberwastaken commited on
Commit
63fa8cc
·
1 Parent(s): 7b4e84e

Made Model work again

Browse files
Files changed (1) hide show
  1. 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=10,
160
  do_sample=True,
161
- temperature=0.5,
162
- top_p=0.9,
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