| { | |
| "architectures": [ | |
| "MambaVisionModel" | |
| ], | |
| "auto_map": { | |
| "AutoConfig": "configuration_mambavision.MambaVisionConfig", | |
| "AutoModel": "modeling_mambavision.MambaVisionModel" | |
| }, | |
| "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 | |
| ] | |
| } | |