nvidia
/

MambaVision-B-21K

Image Classification

Model card Files Files and versions

ahatamiz commited on Mar 24

Commit

e55761e

·

verified ·

1 Parent(s): c9bf43c

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,