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,
|