Update README.md
Browse files
README.md
CHANGED
@@ -94,7 +94,7 @@ sequences = pipeline(
|
|
94 |
temperature=0.1,
|
95 |
top_p=0.95,
|
96 |
num_return_sequences=1,
|
97 |
-
eos_token_id=
|
98 |
max_length=256,
|
99 |
)
|
100 |
for seq in sequences:
|
@@ -200,7 +200,7 @@ sequences = pipeline(
|
|
200 |
temperature=0.1,
|
201 |
top_p=0.95,
|
202 |
num_return_sequences=1,
|
203 |
-
eos_token_id=
|
204 |
max_length=256,
|
205 |
)
|
206 |
for seq in sequences:
|
|
|
94 |
temperature=0.1,
|
95 |
top_p=0.95,
|
96 |
num_return_sequences=1,
|
97 |
+
eos_token_id=tokenizer.eos_token_id,
|
98 |
max_length=256,
|
99 |
)
|
100 |
for seq in sequences:
|
|
|
200 |
temperature=0.1,
|
201 |
top_p=0.95,
|
202 |
num_return_sequences=1,
|
203 |
+
eos_token_id=tokenizer.eos_token_id,
|
204 |
max_length=256,
|
205 |
)
|
206 |
for seq in sequences:
|