fCola commited on
Commit
e74f32d
·
verified ·
1 Parent(s): bfa0650

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -55,8 +55,9 @@ class HfModelWrapper:
55
  "pad_token_id": self.tokenizer.eos_token_id,
56
  "max_length": 32768,
57
  "temperature": 0.1,
58
- "top_p": 0.8,
59
- "repetition_penalty": 1.1,
 
60
  }
61
 
62
  thread = Thread(target=self.model.generate, kwargs=gen_kwargs)
 
55
  "pad_token_id": self.tokenizer.eos_token_id,
56
  "max_length": 32768,
57
  "temperature": 0.1,
58
+ "top_p": 0.9,
59
+ #"repetition_penalty": 1.1,
60
+ "min_length": 5,
61
  }
62
 
63
  thread = Thread(target=self.model.generate, kwargs=gen_kwargs)