Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -126,7 +126,7 @@ def infer(
|
|
126 |
outputs = model.generate(
|
127 |
input_ids=input_ids,
|
128 |
attention_mask=attention_mask,
|
129 |
-
max_new_tokens=
|
130 |
temperature=0.2,
|
131 |
top_p=0.9,
|
132 |
do_sample=True,
|
|
|
126 |
outputs = model.generate(
|
127 |
input_ids=input_ids,
|
128 |
attention_mask=attention_mask,
|
129 |
+
max_new_tokens=65,
|
130 |
temperature=0.2,
|
131 |
top_p=0.9,
|
132 |
do_sample=True,
|