Update README.md (#8)
Browse files- Update README.md (891216915d8dbf3e2d0b4e59277013c0d3114622)
Co-authored-by: Anton Shapkin <jdev8@users.noreply.huggingface.co>
README.md
CHANGED
@@ -43,8 +43,8 @@ input_ids = tokenizer.encode(
|
|
43 |
|
44 |
# Generate
|
45 |
output = model.generate(
|
46 |
-
input_ids, max_length=
|
47 |
-
|
48 |
)
|
49 |
|
50 |
# Decode output
|
|
|
43 |
|
44 |
# Generate
|
45 |
output = model.generate(
|
46 |
+
input_ids, max_length=60, num_return_sequences=1,
|
47 |
+
early_stopping=True, pad_token_id=tokenizer.eos_token_id,
|
48 |
)
|
49 |
|
50 |
# Decode output
|