Rijgersberg commited on
Commit
819b08c
1 Parent(s): 85065a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -87,7 +87,7 @@ def generate(
87
  generate_kwargs = dict(
88
  {"input_ids": input_ids},
89
  streamer=streamer,
90
- max_new_tokens=max_new_tokens,
91
  do_sample=True,
92
  top_p=top_p,
93
  temperature=temperature,
 
87
  generate_kwargs = dict(
88
  {"input_ids": input_ids},
89
  streamer=streamer,
90
+ max_new_tokens=max_tokens,
91
  do_sample=True,
92
  top_p=top_p,
93
  temperature=temperature,