Zimix commited on
Commit
8fbad73
1 Parent(s): 1e519a7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -38,11 +38,11 @@ generation_output = model.generate(**inputs,
38
  # num_beams=5,
39
  eos_token_id=50256,
40
  pad_token_id=0,
41
- # stopping_criteria = StoppingCriteriaList([custom_stopping(stop_token=50256)]),
42
  num_return_sequences = 5)
43
 
44
  for idx,sentence in enumerate(generation_output.sequences):
45
- print('next sentence %d:\n'%idx,tokenizer.decode(sentence).split('<|endoftext|>')[0])
 
46
  print('*'*40)
47
  ```
48
 
 
38
  # num_beams=5,
39
  eos_token_id=50256,
40
  pad_token_id=0,
 
41
  num_return_sequences = 5)
42
 
43
  for idx,sentence in enumerate(generation_output.sequences):
44
+ print('next sentence %d:\n'%idx,
45
+ tokenizer.decode(sentence).split('<|endoftext|>')[0])
46
  print('*'*40)
47
  ```
48