sablab commited on
Commit
c73295d
·
verified ·
1 Parent(s): 28f9563

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,11 +15,11 @@ def format_prompt(message, history):
15
  return prompt
16
 
17
  def generate(
18
- prompt, history, temperature=0.9, max_new_tokens=8000, top_p=0.95, repetition_penalty=1.0,
19
  ):
20
  generate_kwargs = dict(
21
  temperature=0.9,
22
- max_new_tokens=8000,
23
  top_p=0.9,
24
  repetition_penalty=1.0,
25
  do_sample=True,
 
15
  return prompt
16
 
17
  def generate(
18
+ prompt, history, temperature=0.9, max_new_tokens=16000, top_p=0.95, repetition_penalty=1.0,
19
  ):
20
  generate_kwargs = dict(
21
  temperature=0.9,
22
+ max_new_tokens=16000,
23
  top_p=0.9,
24
  repetition_penalty=1.0,
25
  do_sample=True,