Update README.md
Browse files
README.md
CHANGED
@@ -112,12 +112,8 @@ output_ids = model.generate(
|
|
112 |
special_token_ids=[tokenizer.sep_token_id],
|
113 |
token_type_id_sections=model.decoder.config.section_ids,
|
114 |
max_length=256,
|
115 |
-
bos_token_id=tokenizer.bos_token_id,
|
116 |
-
eos_token_id=tokenizer.eos_token_id,
|
117 |
-
pad_token_id=tokenizer.pad_token_id,
|
118 |
num_beams=4,
|
119 |
return_dict_in_generate=True,
|
120 |
-
use_cache=True,
|
121 |
)['sequences']
|
122 |
|
123 |
# Findings and impression section:
|
|
|
112 |
special_token_ids=[tokenizer.sep_token_id],
|
113 |
token_type_id_sections=model.decoder.config.section_ids,
|
114 |
max_length=256,
|
|
|
|
|
|
|
115 |
num_beams=4,
|
116 |
return_dict_in_generate=True,
|
|
|
117 |
)['sequences']
|
118 |
|
119 |
# Findings and impression section:
|