Yuchan5386 commited on
Commit
064c20b
·
verified ·
1 Parent(s): 6ddfcfc

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +3 -3
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.65,
177
- top_k=50, top_p=0.75, typical_p=None,
178
- min_len=20):
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)