Spaces:
Running
on
Zero
Running
on
Zero
dreamerdeo
commited on
Commit
•
123ca1a
1
Parent(s):
6f7c7d8
Update app.py
Browse files
app.py
CHANGED
@@ -54,7 +54,7 @@ def predict(message, history):
|
|
54 |
generate_kwargs = dict(
|
55 |
model_inputs,
|
56 |
streamer=streamer,
|
57 |
-
max_new_tokens=
|
58 |
do_sample=True,
|
59 |
top_p= 0.75,
|
60 |
top_k= 60,
|
|
|
54 |
generate_kwargs = dict(
|
55 |
model_inputs,
|
56 |
streamer=streamer,
|
57 |
+
max_new_tokens=1024,
|
58 |
do_sample=True,
|
59 |
top_p= 0.75,
|
60 |
top_k= 60,
|