Update app.py
Browse files
app.py
CHANGED
|
@@ -88,7 +88,7 @@ def wrap_model(model, tokenizer):
|
|
| 88 |
task="text-generation",
|
| 89 |
temperature=0.5,
|
| 90 |
repetition_penalty=2.1,
|
| 91 |
-
no_repeat_ngram_size=3
|
| 92 |
max_new_tokens=400,
|
| 93 |
num_beams=2,
|
| 94 |
pad_token_id=2,
|
|
|
|
| 88 |
task="text-generation",
|
| 89 |
temperature=0.5,
|
| 90 |
repetition_penalty=2.1,
|
| 91 |
+
no_repeat_ngram_size=3,
|
| 92 |
max_new_tokens=400,
|
| 93 |
num_beams=2,
|
| 94 |
pad_token_id=2,
|