codelion commited on
Commit
9716406
1 Parent(s): 9938eb2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -64,8 +64,8 @@ def generate(
64
  top_p=top_p,
65
  #top_k=top_k,
66
  temperature=temperature,
67
- eos_token_id=pipe.tokenizer.eos_token_id,
68
- pad_token_id=pipe.tokenizer.pad_token_id,
69
  #num_beams=1,
70
  #repetition_penalty=1.2,
71
  )
 
64
  top_p=top_p,
65
  #top_k=top_k,
66
  temperature=temperature,
67
+ eos_token_id=tokenizer.eos_token_id,
68
+ pad_token_id=tokenizer.pad_token_id,
69
  #num_beams=1,
70
  #repetition_penalty=1.2,
71
  )