Chris4K commited on
Commit
2dedea5
·
verified ·
1 Parent(s): 21efff8

Update services/chat_service.py

Browse files
Files changed (1) hide show
  1. services/chat_service.py +1 -1
services/chat_service.py CHANGED
@@ -247,7 +247,7 @@ class ChatService:
247
  do_sample=True,
248
  no_repeat_ngram_size=3,
249
  early_stopping=False, # True num_beams=3, # Increase number of beams if beam search is needed
250
- pad_token_id=tokenizer.eos_token,
251
  )
252
 
253
  response = self.tokenizer.decode(
 
247
  do_sample=True,
248
  no_repeat_ngram_size=3,
249
  early_stopping=False, # True num_beams=3, # Increase number of beams if beam search is needed
250
+ pad_token_id=self.tokenizer.eos_token,
251
  )
252
 
253
  response = self.tokenizer.decode(