Yuchan5386 commited on
Commit
6ddfcfc
·
verified ·
1 Parent(s): 78b0ed5

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +2 -2
api.py CHANGED
@@ -173,8 +173,8 @@ 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.75,
177
- top_k=50, top_p=0.85, typical_p=None,
178
  min_len=20):
179
  model_input = text_to_ids(f"<start> {prompt} <sep>")
180
  model_input = model_input[:max_len]
 
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]