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_variable.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
     attention_mask: torch.FloatTensor
 
 
@@ -86,7 +86,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,
         )
     
 
@@ -223,7 +223,7 @@ class VariableCXREncoderDecoderModel(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:9128f8c8767baf52490e0f1bc882cf80ee14434b4cc4844f433c49272cd9dd95
+oid sha256:fd3d75dceeec1cb112f40a5b8acb031384617714e2e441947f8a1ef3bc5df878
 size 449713809