nvidia
/

MambaVision-T-1K

Image Classification

image-feature-extraction

Model card Files Files and versions

ahatamiz commited on Jul 22, 2024

Commit

edb0c22

·

verified ·

1 Parent(s): 7d42257

Update config.json

Files changed (1) hide show

config.json +2 -1

config.json CHANGED Viewed

@@ -1,6 +1,7 @@
 {
   "architectures": [
-    "MambaVisionModelForImageClassification"
   ],
   "auto_map": {
     "AutoConfig": "configuration_mambavision.MambaVisionConfig",

 {
   "architectures": [
+    "MambaVisionModelForImageClassification",
+    "MambaVisionModel"
   ],
   "auto_map": {
     "AutoConfig": "configuration_mambavision.MambaVisionConfig",