Update README.md
Browse files
README.md
CHANGED
|
@@ -62,7 +62,7 @@ sequences = pipeline(
|
|
| 62 |
temperature=0.1,
|
| 63 |
top_p=0.95,
|
| 64 |
num_return_sequences=1,
|
| 65 |
-
|
| 66 |
max_length=400,
|
| 67 |
add_special_tokens=False
|
| 68 |
)
|
|
|
|
| 62 |
temperature=0.1,
|
| 63 |
top_p=0.95,
|
| 64 |
num_return_sequences=1,
|
| 65 |
+
eos_token_id=tokenizer.eos_token_id,
|
| 66 |
max_length=400,
|
| 67 |
add_special_tokens=False
|
| 68 |
)
|