jmcinern commited on
Commit
b5eb228
·
verified ·
1 Parent(s): 20962a4

Update app.py

Browse files

double checking EOS hypothesis

Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -77,8 +77,7 @@ class ChatBot:
77
  temperature=0.6,
78
  do_sample=True,
79
  return_dict_in_generate=True,
80
- pad_token_id=self.tokenizer.eos_token_id,
81
- eos_token_id="<|im_end|>",
82
  )
83
 
84
  # Decode and clean response, with multiple debugs
 
77
  temperature=0.6,
78
  do_sample=True,
79
  return_dict_in_generate=True,
80
+ pad_token_id=self.tokenizer.eos_token_id
 
81
  )
82
 
83
  # Decode and clean response, with multiple debugs