medxiaorudan commited on
Commit
7f3dac0
1 Parent(s): f443cdc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -62,7 +62,7 @@ sequences = pipeline(
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
  )
 
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
  )