# vit_mae

| | |
|---|---|
| **Model type** | `vit_mae` |
| **Class** | `ViTModel` |
| **Task** | `image-classification` |
| **Source** | `models/vit.py` |

## Description

Vision Transformer for image feature extraction.

## Usage

```bash
mobius build --model <MODEL_ID> output_dir/
```

```python
from mobius import build

model = build("<MODEL_ID>")
```