nestole commited on
Commit
f242578
1 Parent(s): d9f8d20

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -1
run.py CHANGED
@@ -118,7 +118,7 @@ def format_prompt(message):
118
  return prompt
119
 
120
  def responsecritical(
121
- prompt, temperature=0.9, max_new_tokens=500, top_p=0.95, repetition_penalty=1.0,
122
  ):
123
  temperature = float(temperature)
124
  if temperature < 1e-2: temperature = 1e-2
 
118
  return prompt
119
 
120
  def responsecritical(
121
+ prompt, temperature=0.9, max_new_tokens=200, top_p=0.95, repetition_penalty=1.0,
122
  ):
123
  temperature = float(temperature)
124
  if temperature < 1e-2: temperature = 1e-2