Spaces:
Sleeping
Sleeping
Update api.py
Browse files
api.py
CHANGED
@@ -173,9 +173,9 @@ def is_greedy_response_acceptable(text):
|
|
173 |
return True
|
174 |
|
175 |
def generate_text_flex(model, prompt, max_len=100, max_gen=98,
|
176 |
-
repetition_penalty=1.2, temperature=0.
|
177 |
-
top_k=50, top_p=0.
|
178 |
-
min_len=
|
179 |
model_input = text_to_ids(f"<start> {prompt} <sep>")
|
180 |
model_input = model_input[:max_len]
|
181 |
generated = list(model_input)
|
|
|
173 |
return True
|
174 |
|
175 |
def generate_text_flex(model, prompt, max_len=100, max_gen=98,
|
176 |
+
repetition_penalty=1.2, temperature=0.55,
|
177 |
+
top_k=50, top_p=0.70, typical_p=None,
|
178 |
+
min_len=12):
|
179 |
model_input = text_to_ids(f"<start> {prompt} <sep>")
|
180 |
model_input = model_input[:max_len]
|
181 |
generated = list(model_input)
|