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

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +1 -1
api.py CHANGED
@@ -174,7 +174,7 @@ def is_greedy_response_acceptable(text):
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=0.72,
178
  min_len=20):
179
  model_input = text_to_ids(f"<start> {prompt} <sep>")
180
  model_input = model_input[:max_len]
 
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]