omeryentur commited on
Commit
40efd3f
·
verified ·
1 Parent(s): 2a9ec14

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -38,7 +38,10 @@ class LlamaInterface:
38
  "max_length": 512,
39
  "temperature": 0.01,
40
  "do_sample": True,
41
- "top_k": 2,
 
 
 
42
  "top_p": 0.95,
43
  }
44
 
 
38
  "max_length": 512,
39
  "temperature": 0.01,
40
  "do_sample": True,
41
+ "pad_token_id": self.tokenizer.pad_token_id,
42
+ "eos_token_id": self.tokenizer.eos_token_id,
43
+ "num_return_sequences": 1,
44
+ "top_k": 50,
45
  "top_p": 0.95,
46
  }
47