jinymusim commited on
Commit
21fa99c
·
1 Parent(s): 36ed4af

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -42,11 +42,11 @@ out = model.generate(tokenized_poet_start,
42
  num_beams=8,
43
  no_repeat_ngram_size=2,
44
  early_stopping=True,
45
- pad_token_id= tokenizer.pad_token_id
46
  eos_token_id = tokenizer.eos_token_id)
47
 
48
  # Decode Poet
49
- decoded_cont = self.tokenizer.decode(out[0], skip_special_tokens=True)
50
 
51
  print(decoded_cont)
52
  ```
 
42
  num_beams=8,
43
  no_repeat_ngram_size=2,
44
  early_stopping=True,
45
+ pad_token_id= tokenizer.pad_token_id,
46
  eos_token_id = tokenizer.eos_token_id)
47
 
48
  # Decode Poet
49
+ decoded_cont = tokenizer.decode(out[0], skip_special_tokens=True)
50
 
51
  print(decoded_cont)
52
  ```