fCola commited on
Commit
ef4cdd6
·
verified ·
1 Parent(s): 1fde90a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -59,8 +59,8 @@ class HfModelWrapper:
59
  "top_k": 50,
60
  "repetition_penalty": 1.1,
61
  "do_sample": True,
62
- "eos_token_id": tokenizer.eos_token_id,
63
- "pad_token_id": tokenizer.pad_token_id,
64
  }
65
 
66
  thread = Thread(target=self.model.generate, kwargs=gen_kwargs)
 
59
  "top_k": 50,
60
  "repetition_penalty": 1.1,
61
  "do_sample": True,
62
+ "eos_token_id": self.tokenizer.eos_token_id,
63
+ "pad_token_id": self.tokenizer.pad_token_id,
64
  }
65
 
66
  thread = Thread(target=self.model.generate, kwargs=gen_kwargs)