pablo-rf commited on
Commit
1332d4e
1 Parent(s): 897f1d0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -136,7 +136,7 @@ generation = generator(
136
  input_text,
137
  do_sample=True,
138
  top_k=10,
139
- eos_token_id=tokenizer.eos_token_id,
140
  )
141
 
142
  print(f"Result: {generation[0]['generated_text']}")
 
136
  input_text,
137
  do_sample=True,
138
  top_k=10,
139
+ eos_token_id=tokenizer.eos_token_id
140
  )
141
 
142
  print(f"Result: {generation[0]['generated_text']}")