ZibinDong
/

embodiedmae-base

Feature Extraction

Model card Files Files and versions

ZibinDong commited on Jun 26

Commit

af53996

·

verified ·

1 Parent(s): 4b587c2

Upload model

Files changed (2) hide show

config.json +2 -2
model.safetensors +2 -2

config.json CHANGED Viewed

@@ -1,12 +1,12 @@
 {
   "apply_layernorm": true,
   "architectures": [
-    "EmbodiedMAEModel"
   ],
   "attention_probs_dropout_prob": 0.0,
   "auto_map": {
     "AutoConfig": "configuration_embodiedmae.EmbodiedMAEConfig",
-    "AutoModel": "modeling_embodiedmae.EmbodiedMAEModel"
   },
   "decoder_hidden_size": 512,
   "decoder_num_attention_heads": 8,

 {
   "apply_layernorm": true,
   "architectures": [
+    "EmbodiedMAEForMaskedImageModeling"
   ],
   "attention_probs_dropout_prob": 0.0,
   "auto_map": {
     "AutoConfig": "configuration_embodiedmae.EmbodiedMAEConfig",
+    "AutoModelForMaskedImageModeling": "modeling_embodiedmae.EmbodiedMAEForMaskedImageModeling"
   },
   "decoder_hidden_size": 512,
   "decoder_num_attention_heads": 8,

model.safetensors CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:0b3ecb0ce4b1f15a538cb8a732a27d2d904c533c410e13e79ffd92446a2bbb09
+size 523885096