anicolson commited on
Commit
64f04a2
1 Parent(s): 4f1f0a7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -4
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: