{ "architectures": [ "MambaVisionModelForImageClassification" ], "auto_map": { "AutoConfig": "configuration_mambavision.MambaVisionConfig", "AutoModel": "modeling_mambavision.MambaVisionModel", "AutoModelForImageClassification": "modeling_mambavision.MambaVisionModelForImageClassification" }, "depths": [ 3, 3, 10, 5 ], "dim": 128, "drop_path_rate": 0.3, "in_dim": 64, "layer_scale": 1e-05, "layer_scale_conv": null, "mlp_ratio": 4, "model_type": "mambavision", "num_heads": [ 2, 4, 8, 16 ], "torch_dtype": "float32", "transformers_version": "4.36.2", "window_size": [ 8, 8, 14, 7 ] }