Upload model

modelling_longitudinal.py

@@ -25,7 +25,7 @@ class CvtWithProjectionHeadConfig(transformers.CvtConfig):
 
 
 class ModelOutputWithProjectionEmbedding(transformers.modeling_outputs.ModelOutput):
-    projected_last_hidden_state: torch.FloatTensor
+    last_hidden_state: torch.FloatTensor
     attention_mask: torch.FloatTensor
 
 
@@ -89,7 +89,7 @@ class VariableCvtWithProjectionHead(transformers.CvtPreTrainedModel):
             return projection
 
         return ModelOutputWithProjectionEmbedding(
-            projected_last_hidden_state=projection, attention_mask=attention_mask,
+            last_hidden_state=projection, attention_mask=attention_mask,
         )
     
 
@@ -245,7 +245,7 @@ class LongitudinalPromptVariableCXREncoderDecoderModel(VisionEncoderDecoderModel
             decoder_hidden_states=decoder_outputs.hidden_states,
             decoder_attentions=decoder_outputs.attentions,
             cross_attentions=decoder_outputs.cross_attentions,
-            encoder_last_hidden_state=encoder_outputs.projected_last_hidden_state,
+            encoder_last_hidden_state=encoder_outputs.last_hidden_state,
             # encoder_hidden_states=encoder_outputs.hidden_states,
             # encoder_attentions=encoder_outputs.attentions,
         )