Yuchan5386 commited on
Commit
14686d0
·
verified ·
1 Parent(s): c0437ed

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +1 -1
api.py CHANGED
@@ -173,7 +173,7 @@ def is_greedy_response_acceptable(text):
173
  return True
174
 
175
  def generate_text_typical(model, prompt, max_len=100, max_gen=98,
176
- typical_p=0.9, min_len=20):
177
  model_input = text_to_ids(f"<start> {prompt} <sep>")
178
  model_input = model_input[:max_len]
179
  generated = list(model_input)
 
173
  return True
174
 
175
  def generate_text_typical(model, prompt, max_len=100, max_gen=98,
176
+ typical_p=0.5, min_len=20):
177
  model_input = text_to_ids(f"<start> {prompt} <sep>")
178
  model_input = model_input[:max_len]
179
  generated = list(model_input)