config.json

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