ganeshkamath89 commited on
Commit
d50b6fd
1 Parent(s): d5a8634

moving definition to single line

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