Update README.md
Browse files
README.md
CHANGED
@@ -109,7 +109,7 @@ sequences = pipeline(
|
|
109 |
temperature=0.1,
|
110 |
top_p=0.95,
|
111 |
num_return_sequences=1,
|
112 |
-
eos_token_id=
|
113 |
max_length=256,
|
114 |
)
|
115 |
for seq in sequences:
|
@@ -234,7 +234,7 @@ sequences = pipeline(
|
|
234 |
temperature=0.1,
|
235 |
top_p=0.95,
|
236 |
num_return_sequences=1,
|
237 |
-
eos_token_id=
|
238 |
max_length=256,
|
239 |
)
|
240 |
for seq in sequences:
|
|
|
109 |
temperature=0.1,
|
110 |
top_p=0.95,
|
111 |
num_return_sequences=1,
|
112 |
+
eos_token_id=tokenizer.eos_token_id,
|
113 |
max_length=256,
|
114 |
)
|
115 |
for seq in sequences:
|
|
|
234 |
temperature=0.1,
|
235 |
top_p=0.95,
|
236 |
num_return_sequences=1,
|
237 |
+
eos_token_id=tokenizer.eos_token_id,
|
238 |
max_length=256,
|
239 |
)
|
240 |
for seq in sequences:
|