Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -344,7 +344,7 @@ def expand_prompt(prompt):
|
|
344 |
outputs = model.generate(
|
345 |
input_ids=input_ids,
|
346 |
attention_mask=attention_mask,
|
347 |
-
max_new_tokens=
|
348 |
temperature=0.2,
|
349 |
top_p=0.9,
|
350 |
do_sample=True,
|
|
|
344 |
outputs = model.generate(
|
345 |
input_ids=input_ids,
|
346 |
attention_mask=attention_mask,
|
347 |
+
max_new_tokens=1024,
|
348 |
temperature=0.2,
|
349 |
top_p=0.9,
|
350 |
do_sample=True,
|