Update app.py
Browse files
app.py
CHANGED
|
@@ -53,7 +53,7 @@ llm_model = transformers.AutoModelForCausalLM.from_pretrained(
|
|
| 53 |
)
|
| 54 |
|
| 55 |
generation_args = {
|
| 56 |
-
"max_new_tokens":
|
| 57 |
"return_full_text": True,
|
| 58 |
"temperature": 0.0,
|
| 59 |
"do_sample": False,
|
|
|
|
| 53 |
)
|
| 54 |
|
| 55 |
generation_args = {
|
| 56 |
+
"max_new_tokens": 30,
|
| 57 |
"return_full_text": True,
|
| 58 |
"temperature": 0.0,
|
| 59 |
"do_sample": False,
|