Upload model

config.json

@@ -2,6 +2,10 @@
   "architectures": [
     "MambaVisionModelForImageClassification"
   ],
+  "auto_map": {
+    "AutoConfig": "configuration_mambavision.MambaVisionConfig",
+    "AutoModelForImageClassification": "modeling_mambavision.MambaVisionModelForImageClassification"
+  },
   "depths": [
     1,
     3,
@@ -12,6 +16,7 @@
   "drop_path_rate": 0.2,
   "in_dim": 32,
   "mlp_ratio": 4,
+  "model_type": "MambaVision-T-1K",
   "num_heads": [
     2,
     4,

configuration_mambavision.py

@@ -3,7 +3,7 @@ from typing import List
 
 
 class MambaVisionConfig(PretrainedConfig):
-    model_type = "mamba_vision_T"
+    model_type = "MambaVision-T-1K"
 
     def __init__(
         self,
@@ -23,4 +23,15 @@ class MambaVisionConfig(PretrainedConfig):
         self.in_dim = in_dim
         self.mlp_ratio = mlp_ratio
         self.drop_path_rate = drop_path_rate
-        super().__init__(**kwargs)
+        super().__init__(**kwargs)
+
+
+mambavisionT_config = MambaVisionConfig(depths=[1, 3, 8, 4],
+        num_heads=[2, 4, 8, 16],
+        window_size=[8, 8, 14, 7],
+        dim=80,
+        in_dim=32,
+        mlp_ratio=4,
+        drop_path_rate=0.2,)
+
+mambavisionT_config.save_pretrained("MambaVision-T-1K")

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9ae9fdbd5ee3981e4b907aab065abfb0a8d25173b3299c63692291bbfb9b3ecf
+oid sha256:c4e015b2e240dafd14d788f8004b7b62c8d05ab5a2525d1d608f44557a2e93db
 size 127219000