Update README.md
Browse files
README.md
CHANGED
@@ -79,7 +79,7 @@ for data in tqdm(datasets):
|
|
79 |
with torch.no_grad():
|
80 |
outputs = model.generate(
|
81 |
tokenized_input,
|
82 |
-
max_new_tokens=
|
83 |
do_sample=False,
|
84 |
repetition_penalty=1.2
|
85 |
)[0]
|
|
|
79 |
with torch.no_grad():
|
80 |
outputs = model.generate(
|
81 |
tokenized_input,
|
82 |
+
max_new_tokens=512,
|
83 |
do_sample=False,
|
84 |
repetition_penalty=1.2
|
85 |
)[0]
|