Update README.md
Browse files
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,
|
|
|
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 |
|