arabmejo commited on
Commit
c792806
·
verified ·
1 Parent(s): 3c5b4a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def format_prompt(message, history):
15
  prompt += f"[INST] {message} [/INST]"
16
  return prompt
17
 
18
- def generate(prompt, history, temperature=0.1, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0):
19
  temperature = float(temperature)
20
  if temperature < 1e-2:
21
  temperature = 1e-2
 
15
  prompt += f"[INST] {message} [/INST]"
16
  return prompt
17
 
18
+ def generate(prompt, history, temperature=0.1, max_new_tokens=128, top_p=0.95, repetition_penalty=1.0):
19
  temperature = float(temperature)
20
  if temperature < 1e-2:
21
  temperature = 1e-2