ScreenVLM MLX 4-bit
MLX 4-bit quantized version of
docling-project/ScreenVLM for fast inference on Apple Silicon.
Model Details
- Base model: ScreenVLM (316M params, Idefics3 = SigLIP2-base-patch16-512 + Granite 165M)
- Quantization: 4-bit affine (7.654 bits/weight avg, vision encoder at higher precision)
- Size: 288 MB (vs 721 MB original float32)
- License: Apache 2.0
Performance (Apple M4, 64GB)
| Metric | Value |
|---|
| Prompt processing | 747–1382 tok/s |
| Generation speed | 432–462 tok/s |
| Inference time | ~1.7s (172 tokens) |
| Peak memory | 1.1–1.2 GB |
| Model load | 1.1s |
~500× faster than PyTorch CPU on the same hardware.
Usage
1from mlx_vlm import load, generate
2from mlx_vlm.prompt_utils import apply_chat_template
3from mlx_vlm.utils import load_config
4
5model, processor = load("olragon/ScreenVLM-MLX-4bit")
6config = load_config("olragon/ScreenVLM-MLX-4bit")
7
8prompt = apply_chat_template(processor, config, "<screentag>", num_images=1)
9output = generate(model, processor, prompt, image="screenshot.png", max_tokens=2048)
10print(output)
Output Format (ScreenTag)
55 UI element classes with normalized bounding boxes (0–500 grid):
1<button><loc_391><loc_46><loc_451><loc_49>Get started</button>
2<tab><loc_582><loc_170><loc_633><loc_174>Tables</tab>
3<logo><loc_62><loc_19><loc_182><loc_42>filament</logo>
4<text><loc_73><loc_171><loc_427><loc_175>A cohesive set of building blocks</text>
Element types include: Button, Navigation Bar, Text Input, Link, Tab, Image, Video, Table, List, Card, Badge, Avatar, Alert, Search Bar, Logo, Heading, Code snippet, Checkbox, and more.
Conversion
1python -m mlx_vlm.convert \
2 --model docling-project/ScreenVLM \
3 --quantize --q-bits 4 \
4 --mlx-path ./ScreenVLM-MLX-4bit
Requires mlx-vlm >= 0.1.12, torch, torchvision (for image processor conversion).
Citation
1@inproceedings{gurbuz2026screenparse,
2 title={ScreenParse: Moving Beyond Sparse Grounding with Complete Screen Parsing Supervision},
3 author={Gurbuz, A. Said and Hong, Sunghwan and Nassar, Ahmed and Pollefeys, Marc and Staar, Peter},
4 booktitle={ICML},
5 year={2026}
6}
Acknowledgments
Original model by
IBM Research & ETH Zurich. MLX conversion by
olragon.