medxiaorudan
commited on
Commit
•
7f3dac0
1
Parent(s):
f443cdc
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 |
+
pad_token_id=tokenizer.eos_token_id,
|
66 |
max_length=400,
|
67 |
add_special_tokens=False
|
68 |
)
|