alfredplpl commited on
Commit
8f2511c
1 Parent(s): e9d3e3a

Update README.md

Browse files

tokenizer.eos_token_id is not defined. So, this example cannot run.
I recommend remove the line temporally.

Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -85,8 +85,7 @@ with torch.no_grad():
85
  temperature=0.5,
86
  top_p=0.9,
87
  pad_token_id=tokenizer.pad_token_id,
88
- bos_token_id=tokenizer.bos_token_id,
89
- eos_token_id=tokenizer.eos_token_id
90
  )
91
 
92
  output = tokenizer.decode(output_ids.tolist()[0])
 
85
  temperature=0.5,
86
  top_p=0.9,
87
  pad_token_id=tokenizer.pad_token_id,
88
+ bos_token_id=tokenizer.bos_token_id
 
89
  )
90
 
91
  output = tokenizer.decode(output_ids.tolist()[0])