Upload model

config.json

@@ -78,7 +78,7 @@
     "top_p": 1.0,
     "torch_dtype": null,
     "torchscript": false,
-    "transformers_version": "4.29.2",
+    "transformers_version": "4.28.1",
     "type_vocab_size": 2,
     "typical_p": 1.0,
     "use_bfloat16": false,
@@ -2243,7 +2243,7 @@
     "top_p": 1.0,
     "torch_dtype": "float32",
     "torchscript": false,
-    "transformers_version": "4.29.2",
+    "transformers_version": "4.28.1",
     "typical_p": 1.0,
     "use_bfloat16": false
   },

generation_config.json

@@ -1,5 +1,5 @@
 {
   "_from_model_config": true,
   "pad_token_id": 0,
-  "transformers_version": "4.29.2"
+  "transformers_version": "4.28.1"
 }

modelling_single.py

@@ -22,7 +22,7 @@ class CvtWithProjectionHeadConfig(transformers.CvtConfig):
 
 
 class ModelOutputWithProjectionEmbedding(transformers.modeling_outputs.ModelOutput):
-    projected_last_hidden_state: torch.FloatTensor
+    last_hidden_state: torch.FloatTensor
 
 
 class CvtProjectionHead(torch.nn.Module):
@@ -76,7 +76,7 @@ class CvtWithProjectionHead(transformers.CvtPreTrainedModel):
             return projection
 
         return ModelOutputWithProjectionEmbedding(
-            projected_last_hidden_state=projection,
+            last_hidden_state=projection,
         )
     
 
@@ -210,7 +210,7 @@ class SingleCXREncoderDecoderModel(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,
         )

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1d111366fdd206e51232d2f206f51de7e0707da9658a5237c694450b94935698
+oid sha256:e1b86278e3c7a3dcf77f4a5804f25eaaba4de4ccd85f09a87813843a6d54a87e
 size 449713809