Upload folder using huggingface_hub

modelling_medicap.py

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