Spaces:
Sleeping
Sleeping
Update api.py
Browse files
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 |
|