{ "architectures": [ "MambaVisionModelForImageClassification" ], "auto_map": { "AutoConfig": ".configuration_mambavision.MambaVisionConfig", "AutoModelForImageClassification": ".modeling_mambavision.MambaVisionModelForImageClassification" }, "depths": [ 1, 3, 8, 4 ], "dim": 80, "drop_path_rate": 0.2, "in_dim": 32, "mlp_ratio": 4, "model_type": "MambaVision-T-1K", "num_heads": [ 2, 4, 8, 16 ], "torch_dtype": "float32", "transformers_version": "4.36.2", "window_size": [ 8, 8, 14, 7 ] }