Spaces:
Runtime error
Runtime error
Rijgersberg
commited on
Commit
•
819b08c
1
Parent(s):
85065a5
Update app.py
Browse files
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=
|
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,
|