itsanurag commited on
Commit
dc2af74
·
verified ·
1 Parent(s): e3affdc

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, system_prompt, temperature=0.9, 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, system_prompt, temperature=0.9, max_new_tokens=180, top_p=0.95, repetition_penalty=1.0):
19
  temperature = float(temperature)
20
  if temperature < 1e-2:
21
  temperature = 1e-2