winglian commited on
Commit
f111d2b
1 Parent(s): 8e1ed27

change generation config a bit

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -57,9 +57,9 @@ class Pipeline:
57
  self.generation_config = {
58
  "max_new_tokens": 1024,
59
  "top_k": 40,
60
- "top_p": 0.95,
61
- "temperature": 0.8,
62
- "repetition_penalty": 1.1,
63
  "last_n_tokens": 64,
64
  "seed": -1,
65
  "batch_size": 8,
 
57
  self.generation_config = {
58
  "max_new_tokens": 1024,
59
  "top_k": 40,
60
+ "top_p": 0.90,
61
+ "temperature": 0.72,
62
+ "repetition_penalty": 1.22,
63
  "last_n_tokens": 64,
64
  "seed": -1,
65
  "batch_size": 8,