Crystalcareai commited on
Commit
7bc62e5
·
verified ·
1 Parent(s): dd18e79

Update generate.py

Browse files
Files changed (1) hide show
  1. generate.py +2 -5
generate.py CHANGED
@@ -228,9 +228,6 @@ def generate(
228
  **model_kwargs,
229
  )
230
 
231
- generated_text = self.tokenizer.decode(generated_token_ids[0], skip_special_tokens=True)
232
 
233
- if dynamic_temperature is not None:
234
- return generated_text
235
- else:
236
- return generated_token_ids, generated_text
 
228
  **model_kwargs,
229
  )
230
 
 
231
 
232
+ generated_text = self.tokenizer.decode(generated_token_ids[0], skip_special_tokens=False)
233
+ return generated_token_ids, generated_text