nestole commited on
Commit
0aec7e8
·
verified ·
1 Parent(s): bd582eb

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +2 -2
run.py CHANGED
@@ -117,7 +117,7 @@ def format_prompt(message):
117
  def responsecritical(
118
  prompt, temperature=0.9, max_new_tokens=500, top_p=0.95, repetition_penalty=1.0,
119
  ):
120
- temperature = float(temperature)
121
  if temperature < 1e-2: temperature = 1e-2
122
  top_p = float(top_p)
123
  generate_kwargs = dict(
@@ -146,7 +146,7 @@ def responsecritical(
146
  def response(
147
  prompt, temperature=0.9, max_new_tokens=500, top_p=0.95, repetition_penalty=1.0,
148
  ):
149
- temperature = float(temperature)
150
  if temperature < 1e-2: temperature = 1e-2
151
  top_p = float(top_p)
152
  generate_kwargs = dict(
 
117
  def responsecritical(
118
  prompt, temperature=0.9, max_new_tokens=500, top_p=0.95, repetition_penalty=1.0,
119
  ):
120
+ temperature = float(0.9)
121
  if temperature < 1e-2: temperature = 1e-2
122
  top_p = float(top_p)
123
  generate_kwargs = dict(
 
146
  def response(
147
  prompt, temperature=0.9, max_new_tokens=500, top_p=0.95, repetition_penalty=1.0,
148
  ):
149
+ temperature = float(0.9)
150
  if temperature < 1e-2: temperature = 1e-2
151
  top_p = float(top_p)
152
  generate_kwargs = dict(