johann22 commited on
Commit
f4800cd
1 Parent(s): 8a306f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ def run_gpt(
57
 
58
  generate_kwargs = dict(
59
  temperature=0.9,
60
- max_new_tokens=10480,
61
  top_p=0.95,
62
  repetition_penalty=1.0,
63
  do_sample=True,
 
57
 
58
  generate_kwargs = dict(
59
  temperature=0.9,
60
+ max_new_tokens=1048,
61
  top_p=0.95,
62
  repetition_penalty=1.0,
63
  do_sample=True,