nvidia
/

MambaVision-T-1K

Image Classification

image-feature-extraction

Model card Files Files and versions

ahatamiz commited on Jul 22, 2024

Commit

774ef74

·

verified ·

1 Parent(s): e7e0c47

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -3,8 +3,8 @@
     "MambaVisionModelForImageClassification"
   ],
   "auto_map": {
-    "AutoConfig": "configuration_mambavision.MambaVisionConfig",
-    "AutoModelForImageClassification": "modeling_mambavision.MambaVisionModelForImageClassification"
   },
   "depths": [
     1,

     "MambaVisionModelForImageClassification"
   ],
   "auto_map": {
+    "AutoConfig": ".configuration_mambavision.MambaVisionConfig",
+    "AutoModelForImageClassification": ".modeling_mambavision.MambaVisionModelForImageClassification"
   },
   "depths": [
     1,