Yuchan5386 commited on
Commit
0308d2c
·
verified ·
1 Parent(s): 711e031

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +1 -1
api.py CHANGED
@@ -191,7 +191,7 @@ def generate_text_mirostat_top_p(model, prompt, max_len=100, max_gen=98,
191
 
192
  decoded_text = decoded_text.strip()
193
 
194
- if len(generated) >= min_len and (final_token == end_id or decoded_text.endswith(('.', '!', '?'))):
195
  yield decoded_text
196
  break
197
 
 
191
 
192
  decoded_text = decoded_text.strip()
193
 
194
+ if len(generated) >= min_len and (final_token == end_id or decoded_text.endswith(('.', '!', '?', '<end>'))):
195
  yield decoded_text
196
  break
197