anicolson commited on
Commit
62db3d9
1 Parent(s): 65cbcfd

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. modelling_medicap.py +1 -1
modelling_medicap.py CHANGED
@@ -247,7 +247,7 @@ class MedICapEncoderDecoderModel(VisionEncoderDecoderModel):
247
  caption = [f'{tokenizer.bos_token}{i}{tokenizer.eos_token}' for i in captions]
248
 
249
  # Tokenize the caption:
250
- tokenized = self.tokenizer(
251
  caption,
252
  padding='longest',
253
  truncation=True,
 
247
  caption = [f'{tokenizer.bos_token}{i}{tokenizer.eos_token}' for i in captions]
248
 
249
  # Tokenize the caption:
250
+ tokenized = tokenizer(
251
  caption,
252
  padding='longest',
253
  truncation=True,