Upload folder using huggingface_hub

modelling_medicap.py

@@ -176,13 +176,14 @@ class MedICapEncoderDecoderModel(VisionEncoderDecoderModel):
         embeddings = self.decoder.transformer.wte(decoder_input_ids)
         embeddings = torch.cat([encoder_outputs[0], embeddings], dim=1)
 
-        decoder_attention_mask = torch.cat(
-            [
-                torch.ones(encoder_outputs[0].shape[:-1], dtype=decoder_attention_mask.dtype, device=self.device), 
-                decoder_attention_mask
-            ], 
-            dim=1,
-        )
+        if decoder_attention_mask:
+            decoder_attention_mask = torch.cat(
+                [
+                    torch.ones(encoder_outputs[0].shape[:-1], dtype=decoder_attention_mask.dtype, device=self.device), 
+                    decoder_attention_mask
+                ], 
+                dim=1,
+            )
         
         decoder_outputs = self.decoder(
             input_ids=decoder_input_ids,