cochi1706 commited on
Commit
d9b7716
·
1 Parent(s): ba4780e

Remove eos_token_id and pad_token_id from text generation parameters in chatbot application to streamline configuration and maintain focus on essential settings.

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -24,9 +24,7 @@ text_generator = pipeline(
24
  device=device,
25
  max_length=520,
26
  do_sample=True,
27
- temperature=0.7,
28
- eos_token_id=tokenizer.eos_token_id, # Dừng khi gặp EOS token
29
- pad_token_id=tokenizer.pad_token_id, # Sử dụng pad token
30
  )
31
 
32
  model.eval()
 
24
  device=device,
25
  max_length=520,
26
  do_sample=True,
27
+ temperature=0.7
 
 
28
  )
29
 
30
  model.eval()