Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -90,7 +90,7 @@ if generate_btn and user_prompt.strip():
|
|
| 90 |
input_ids = tokenizer(gen_prompt, return_tensors="pt").input_ids
|
| 91 |
output_ids = text_model.generate(
|
| 92 |
input_ids,
|
| 93 |
-
max_new_tokens=
|
| 94 |
temperature=0.9,
|
| 95 |
top_p=0.95,
|
| 96 |
do_sample=True,
|
|
|
|
| 90 |
input_ids = tokenizer(gen_prompt, return_tensors="pt").input_ids
|
| 91 |
output_ids = text_model.generate(
|
| 92 |
input_ids,
|
| 93 |
+
max_new_tokens=500,
|
| 94 |
temperature=0.9,
|
| 95 |
top_p=0.95,
|
| 96 |
do_sample=True,
|