Upload model

configuration_mambavision.py

@@ -23,15 +23,4 @@ class MambaVisionConfig(PretrainedConfig):
         self.in_dim = in_dim
         self.mlp_ratio = mlp_ratio
         self.drop_path_rate = drop_path_rate
-        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")
+        super().__init__(**kwargs)

model.safetensors

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

modeling_mambavision.py

@@ -28,7 +28,10 @@ from einops import rearrange, repeat
 
 from transformers import PreTrainedModel
 
-from configuration_mambavision import MambaVisionConfig
+try:
+    from .configuration_mambavision import MambaVisionConfig
+except:
+    from configuration_mambavision import MambaVisionConfig
 
 
 def _cfg(url='', **kwargs):