hsuwill000
commited on
Commit
•
c48b974
1
Parent(s):
c3337cd
Update app.py
Browse files
app.py
CHANGED
@@ -36,7 +36,7 @@ def predict(message, history):
|
|
36 |
generate_kwargs = dict(
|
37 |
model_inputs,
|
38 |
streamer=streamer,
|
39 |
-
max_new_tokens=
|
40 |
do_sample=True,
|
41 |
top_p=0.95,
|
42 |
top_k=50,
|
|
|
36 |
generate_kwargs = dict(
|
37 |
model_inputs,
|
38 |
streamer=streamer,
|
39 |
+
max_new_tokens=512,
|
40 |
do_sample=True,
|
41 |
top_p=0.95,
|
42 |
top_k=50,
|