nvidia
/

MambaVision-B-1K

Image Feature Extraction

image-classification

Model card Files Files and versions Community

ahatamiz commited on Jul 24

Commit

4ae3b96

•

1 Parent(s): 54acb42

Upload model

Files changed (1) hide show

config.json +3 -2

config.json CHANGED Viewed

@@ -1,10 +1,11 @@
 {
   "architectures": [
-    "MambaVisionModel"
   ],
   "auto_map": {
     "AutoConfig": "configuration_mambavision.MambaVisionConfig",
-    "AutoModel": "modeling_mambavision.MambaVisionModel"
   },
   "depths": [
     3,

 {
   "architectures": [
+    "MambaVisionModelForImageClassification"
   ],
   "auto_map": {
     "AutoConfig": "configuration_mambavision.MambaVisionConfig",
+    "AutoModel": "modeling_mambavision.MambaVisionModel",
+    "AutoModelForImageClassification": "modeling_mambavision.MambaVisionModelForImageClassification"
   },
   "depths": [
     3,