config.json

{
  "architectures": [
    "MambaVisionModelForImageClassification"
  ],
  "auto_map": {
    "AutoConfig": "configuration_mambavision.MambaVisionConfig",
    "AutoModel": "modeling_mambavision.MambaVisionModel",
    "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",
  "num_heads": [
    2,
    4,
    8,
    16
  ],
  "torch_dtype": "float32",
  "transformers_version": "4.36.2",
  "window_size": [
    8,
    8,
    14,
    7
  ]
}