aehrc
/

cxrmate-ed

Model card Files Files and versions Community

anicolson commited on 3 days ago

Commit

51e91da

•

1 Parent(s): e39d37c

Upload model

Files changed (3) hide show

config.json +2 -1
configuration_cxrmate_ed.py +1 -1
model.safetensors +1 -1

config.json CHANGED Viewed

@@ -3,6 +3,7 @@
     "CXRMateEDModel"
   ],
   "auto_map": {
     "AutoModel": "modelling_cxrmate_ed.CXRMateEDModel"
   },
   "decoder": {
@@ -35,7 +36,7 @@
     "projection_size": 768
   },
   "is_encoder_decoder": false,
-  "model_type": "vision-encoder-decoder",
   "tie_word_embeddings": false,
   "torch_dtype": "float32",
   "transformers_version": "4.39.3"

     "CXRMateEDModel"
   ],
   "auto_map": {
+    "AutoConfig": "configuration_cxrmate_ed.CXRMateEDConfig",
     "AutoModel": "modelling_cxrmate_ed.CXRMateEDModel"
   },
   "decoder": {
     "projection_size": 768
   },
   "is_encoder_decoder": false,
+  "model_type": "cxrmate-ed",
   "tie_word_embeddings": false,
   "torch_dtype": "float32",
   "transformers_version": "4.39.3"

configuration_cxrmate_ed.py CHANGED Viewed

@@ -9,7 +9,7 @@ logger = logging.get_logger(__name__)
 class CXRMateEDConfig(PretrainedConfig):
-    model_type = "vision-encoder-decoder"
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

 class CXRMateEDConfig(PretrainedConfig):
+    model_type = "cxrmate-ed"
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

model.safetensors CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:60f9f0d9dca5f3da1720bed932a4770d2135919a759d7843a5911cf6ef9b25f3
 size 789958760