Upload folder using huggingface_hub
Browse files- 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 =
|
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,
|