Update README.md
Browse files
README.md
CHANGED
|
@@ -135,7 +135,7 @@ input_ids = tokenizer(
|
|
| 135 |
|
| 136 |
output_ids = model.generate(
|
| 137 |
input_ids=input_ids,
|
| 138 |
-
max_length=
|
| 139 |
no_repeat_ngram_size=2,
|
| 140 |
num_beams=4
|
| 141 |
)[0]
|
|
|
|
| 135 |
|
| 136 |
output_ids = model.generate(
|
| 137 |
input_ids=input_ids,
|
| 138 |
+
max_length=1000,
|
| 139 |
no_repeat_ngram_size=2,
|
| 140 |
num_beams=4
|
| 141 |
)[0]
|