Image-Text-to-Text
MLX
Safetensors
unlimited-ocr
ax-engine
mlx-vlm
ocr
mxfp8
int8
apple-silicon
automatosx
conversational
8-bit precision
Instructions to use AutomatosX/AX-Unlimited-OCR-3B-MoE-MLX-MXFP8 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- MLX
How to use AutomatosX/AX-Unlimited-OCR-3B-MoE-MLX-MXFP8 with MLX:
# Make sure mlx-vlm is installed # pip install --upgrade mlx-vlm from mlx_vlm import load, generate from mlx_vlm.prompt_utils import apply_chat_template from mlx_vlm.utils import load_config # Load the model model, processor = load("AutomatosX/AX-Unlimited-OCR-3B-MoE-MLX-MXFP8") config = load_config("AutomatosX/AX-Unlimited-OCR-3B-MoE-MLX-MXFP8") # Prepare input image = ["http://images.cocodataset.org/val2017/000000039769.jpg"] prompt = "Describe this image." # Apply chat template formatted_prompt = apply_chat_template( processor, config, prompt, num_images=1 ) # Generate output output = generate(model, processor, formatted_prompt, image) print(output) - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- LM Studio
- Atomic Chat
| { | |
| "method": "mxfp8", | |
| "group_size": 32, | |
| "bits": 8, | |
| "source_model": "Unlimited-OCR", | |
| "source_revision": "ee63731b6461c8afcdcc7b15352e7d2ffecc2ead", | |
| "precision_map_sha256": "c5f8b260d2e1ff2fc0405c6ef3857829ca85f49b1980a8c599fb03e06788eb1a", | |
| "quantized_module_count": 119, | |
| "quantized_precision_counts": { | |
| "affine8": 1, | |
| "mxfp8": 118 | |
| }, | |
| "preserved_quantizable_module_count": 146, | |
| "quantized_modules": [ | |
| "language_model.lm_head", | |
| "language_model.model.embed_tokens", | |
| "language_model.model.layers.0.mlp.down_proj", | |
| "language_model.model.layers.0.mlp.gate_proj", | |
| "language_model.model.layers.0.mlp.up_proj", | |
| "language_model.model.layers.0.self_attn.k_proj", | |
| "language_model.model.layers.0.self_attn.o_proj", | |
| "language_model.model.layers.0.self_attn.q_proj", | |
| "language_model.model.layers.0.self_attn.v_proj", | |
| "language_model.model.layers.1.mlp.shared_experts.down_proj", | |
| "language_model.model.layers.1.mlp.shared_experts.gate_proj", | |
| "language_model.model.layers.1.mlp.shared_experts.up_proj", | |
| "language_model.model.layers.1.mlp.switch_mlp.down_proj", | |
| "language_model.model.layers.1.mlp.switch_mlp.gate_proj", | |
| "language_model.model.layers.1.mlp.switch_mlp.up_proj", | |
| "language_model.model.layers.1.self_attn.k_proj", | |
| "language_model.model.layers.1.self_attn.o_proj", | |
| "language_model.model.layers.1.self_attn.q_proj", | |
| "language_model.model.layers.1.self_attn.v_proj", | |
| "language_model.model.layers.10.mlp.shared_experts.down_proj", | |
| "language_model.model.layers.10.mlp.shared_experts.gate_proj", | |
| "language_model.model.layers.10.mlp.shared_experts.up_proj", | |
| "language_model.model.layers.10.mlp.switch_mlp.down_proj", | |
| "language_model.model.layers.10.mlp.switch_mlp.gate_proj", | |
| "language_model.model.layers.10.mlp.switch_mlp.up_proj", | |
| "language_model.model.layers.10.self_attn.k_proj", | |
| "language_model.model.layers.10.self_attn.o_proj", | |
| "language_model.model.layers.10.self_attn.q_proj", | |
| "language_model.model.layers.10.self_attn.v_proj", | |
| "language_model.model.layers.11.mlp.shared_experts.down_proj", | |
| "language_model.model.layers.11.mlp.shared_experts.gate_proj", | |
| "language_model.model.layers.11.mlp.shared_experts.up_proj", | |
| "language_model.model.layers.11.mlp.switch_mlp.down_proj", | |
| "language_model.model.layers.11.mlp.switch_mlp.gate_proj", | |
| "language_model.model.layers.11.mlp.switch_mlp.up_proj", | |
| "language_model.model.layers.11.self_attn.k_proj", | |
| "language_model.model.layers.11.self_attn.o_proj", | |
| "language_model.model.layers.11.self_attn.q_proj", | |
| "language_model.model.layers.11.self_attn.v_proj", | |
| "language_model.model.layers.2.mlp.shared_experts.down_proj", | |
| "language_model.model.layers.2.mlp.shared_experts.gate_proj", | |
| "language_model.model.layers.2.mlp.shared_experts.up_proj", | |
| "language_model.model.layers.2.mlp.switch_mlp.down_proj", | |
| "language_model.model.layers.2.mlp.switch_mlp.gate_proj", | |
| "language_model.model.layers.2.mlp.switch_mlp.up_proj", | |
| "language_model.model.layers.2.self_attn.k_proj", | |
| "language_model.model.layers.2.self_attn.o_proj", | |
| "language_model.model.layers.2.self_attn.q_proj", | |
| "language_model.model.layers.2.self_attn.v_proj", | |
| "language_model.model.layers.3.mlp.shared_experts.down_proj", | |
| "language_model.model.layers.3.mlp.shared_experts.gate_proj", | |
| "language_model.model.layers.3.mlp.shared_experts.up_proj", | |
| "language_model.model.layers.3.mlp.switch_mlp.down_proj", | |
| "language_model.model.layers.3.mlp.switch_mlp.gate_proj", | |
| "language_model.model.layers.3.mlp.switch_mlp.up_proj", | |
| "language_model.model.layers.3.self_attn.k_proj", | |
| "language_model.model.layers.3.self_attn.o_proj", | |
| "language_model.model.layers.3.self_attn.q_proj", | |
| "language_model.model.layers.3.self_attn.v_proj", | |
| "language_model.model.layers.4.mlp.shared_experts.down_proj", | |
| "language_model.model.layers.4.mlp.shared_experts.gate_proj", | |
| "language_model.model.layers.4.mlp.shared_experts.up_proj", | |
| "language_model.model.layers.4.mlp.switch_mlp.down_proj", | |
| "language_model.model.layers.4.mlp.switch_mlp.gate_proj", | |
| "language_model.model.layers.4.mlp.switch_mlp.up_proj", | |
| "language_model.model.layers.4.self_attn.k_proj", | |
| "language_model.model.layers.4.self_attn.o_proj", | |
| "language_model.model.layers.4.self_attn.q_proj", | |
| "language_model.model.layers.4.self_attn.v_proj", | |
| "language_model.model.layers.5.mlp.shared_experts.down_proj", | |
| "language_model.model.layers.5.mlp.shared_experts.gate_proj", | |
| "language_model.model.layers.5.mlp.shared_experts.up_proj", | |
| "language_model.model.layers.5.mlp.switch_mlp.down_proj", | |
| "language_model.model.layers.5.mlp.switch_mlp.gate_proj", | |
| "language_model.model.layers.5.mlp.switch_mlp.up_proj", | |
| "language_model.model.layers.5.self_attn.k_proj", | |
| "language_model.model.layers.5.self_attn.o_proj", | |
| "language_model.model.layers.5.self_attn.q_proj", | |
| "language_model.model.layers.5.self_attn.v_proj", | |
| "language_model.model.layers.6.mlp.shared_experts.down_proj", | |
| "language_model.model.layers.6.mlp.shared_experts.gate_proj", | |
| "language_model.model.layers.6.mlp.shared_experts.up_proj", | |
| "language_model.model.layers.6.mlp.switch_mlp.down_proj", | |
| "language_model.model.layers.6.mlp.switch_mlp.gate_proj", | |
| "language_model.model.layers.6.mlp.switch_mlp.up_proj", | |
| "language_model.model.layers.6.self_attn.k_proj", | |
| "language_model.model.layers.6.self_attn.o_proj", | |
| "language_model.model.layers.6.self_attn.q_proj", | |
| "language_model.model.layers.6.self_attn.v_proj", | |
| "language_model.model.layers.7.mlp.shared_experts.down_proj", | |
| "language_model.model.layers.7.mlp.shared_experts.gate_proj", | |
| "language_model.model.layers.7.mlp.shared_experts.up_proj", | |
| "language_model.model.layers.7.mlp.switch_mlp.down_proj", | |
| "language_model.model.layers.7.mlp.switch_mlp.gate_proj", | |
| "language_model.model.layers.7.mlp.switch_mlp.up_proj", | |
| "language_model.model.layers.7.self_attn.k_proj", | |
| "language_model.model.layers.7.self_attn.o_proj", | |
| "language_model.model.layers.7.self_attn.q_proj", | |
| "language_model.model.layers.7.self_attn.v_proj", | |
| "language_model.model.layers.8.mlp.shared_experts.down_proj", | |
| "language_model.model.layers.8.mlp.shared_experts.gate_proj", | |
| "language_model.model.layers.8.mlp.shared_experts.up_proj", | |
| "language_model.model.layers.8.mlp.switch_mlp.down_proj", | |
| "language_model.model.layers.8.mlp.switch_mlp.gate_proj", | |
| "language_model.model.layers.8.mlp.switch_mlp.up_proj", | |
| "language_model.model.layers.8.self_attn.k_proj", | |
| "language_model.model.layers.8.self_attn.o_proj", | |
| "language_model.model.layers.8.self_attn.q_proj", | |
| "language_model.model.layers.8.self_attn.v_proj", | |
| "language_model.model.layers.9.mlp.shared_experts.down_proj", | |
| "language_model.model.layers.9.mlp.shared_experts.gate_proj", | |
| "language_model.model.layers.9.mlp.shared_experts.up_proj", | |
| "language_model.model.layers.9.mlp.switch_mlp.down_proj", | |
| "language_model.model.layers.9.mlp.switch_mlp.gate_proj", | |
| "language_model.model.layers.9.mlp.switch_mlp.up_proj", | |
| "language_model.model.layers.9.self_attn.k_proj", | |
| "language_model.model.layers.9.self_attn.o_proj", | |
| "language_model.model.layers.9.self_attn.q_proj", | |
| "language_model.model.layers.9.self_attn.v_proj" | |
| ], | |
| "quantized_module_precisions": { | |
| "language_model.lm_head": "affine8", | |
| "language_model.model.embed_tokens": "mxfp8", | |
| "language_model.model.layers.0.mlp.down_proj": "mxfp8", | |
| "language_model.model.layers.0.mlp.gate_proj": "mxfp8", | |
| "language_model.model.layers.0.mlp.up_proj": "mxfp8", | |
| "language_model.model.layers.0.self_attn.k_proj": "mxfp8", | |
| "language_model.model.layers.0.self_attn.o_proj": "mxfp8", | |
| "language_model.model.layers.0.self_attn.q_proj": "mxfp8", | |
| "language_model.model.layers.0.self_attn.v_proj": "mxfp8", | |
| "language_model.model.layers.1.mlp.shared_experts.down_proj": "mxfp8", | |
| "language_model.model.layers.1.mlp.shared_experts.gate_proj": "mxfp8", | |
| "language_model.model.layers.1.mlp.shared_experts.up_proj": "mxfp8", | |
| "language_model.model.layers.1.mlp.switch_mlp.down_proj": "mxfp8", | |
| "language_model.model.layers.1.mlp.switch_mlp.gate_proj": "mxfp8", | |
| "language_model.model.layers.1.mlp.switch_mlp.up_proj": "mxfp8", | |
| "language_model.model.layers.1.self_attn.k_proj": "mxfp8", | |
| "language_model.model.layers.1.self_attn.o_proj": "mxfp8", | |
| "language_model.model.layers.1.self_attn.q_proj": "mxfp8", | |
| "language_model.model.layers.1.self_attn.v_proj": "mxfp8", | |
| "language_model.model.layers.10.mlp.shared_experts.down_proj": "mxfp8", | |
| "language_model.model.layers.10.mlp.shared_experts.gate_proj": "mxfp8", | |
| "language_model.model.layers.10.mlp.shared_experts.up_proj": "mxfp8", | |
| "language_model.model.layers.10.mlp.switch_mlp.down_proj": "mxfp8", | |
| "language_model.model.layers.10.mlp.switch_mlp.gate_proj": "mxfp8", | |
| "language_model.model.layers.10.mlp.switch_mlp.up_proj": "mxfp8", | |
| "language_model.model.layers.10.self_attn.k_proj": "mxfp8", | |
| "language_model.model.layers.10.self_attn.o_proj": "mxfp8", | |
| "language_model.model.layers.10.self_attn.q_proj": "mxfp8", | |
| "language_model.model.layers.10.self_attn.v_proj": "mxfp8", | |
| "language_model.model.layers.11.mlp.shared_experts.down_proj": "mxfp8", | |
| "language_model.model.layers.11.mlp.shared_experts.gate_proj": "mxfp8", | |
| "language_model.model.layers.11.mlp.shared_experts.up_proj": "mxfp8", | |
| "language_model.model.layers.11.mlp.switch_mlp.down_proj": "mxfp8", | |
| "language_model.model.layers.11.mlp.switch_mlp.gate_proj": "mxfp8", | |
| "language_model.model.layers.11.mlp.switch_mlp.up_proj": "mxfp8", | |
| "language_model.model.layers.11.self_attn.k_proj": "mxfp8", | |
| "language_model.model.layers.11.self_attn.o_proj": "mxfp8", | |
| "language_model.model.layers.11.self_attn.q_proj": "mxfp8", | |
| "language_model.model.layers.11.self_attn.v_proj": "mxfp8", | |
| "language_model.model.layers.2.mlp.shared_experts.down_proj": "mxfp8", | |
| "language_model.model.layers.2.mlp.shared_experts.gate_proj": "mxfp8", | |
| "language_model.model.layers.2.mlp.shared_experts.up_proj": "mxfp8", | |
| "language_model.model.layers.2.mlp.switch_mlp.down_proj": "mxfp8", | |
| "language_model.model.layers.2.mlp.switch_mlp.gate_proj": "mxfp8", | |
| "language_model.model.layers.2.mlp.switch_mlp.up_proj": "mxfp8", | |
| "language_model.model.layers.2.self_attn.k_proj": "mxfp8", | |
| "language_model.model.layers.2.self_attn.o_proj": "mxfp8", | |
| "language_model.model.layers.2.self_attn.q_proj": "mxfp8", | |
| "language_model.model.layers.2.self_attn.v_proj": "mxfp8", | |
| "language_model.model.layers.3.mlp.shared_experts.down_proj": "mxfp8", | |
| "language_model.model.layers.3.mlp.shared_experts.gate_proj": "mxfp8", | |
| "language_model.model.layers.3.mlp.shared_experts.up_proj": "mxfp8", | |
| "language_model.model.layers.3.mlp.switch_mlp.down_proj": "mxfp8", | |
| "language_model.model.layers.3.mlp.switch_mlp.gate_proj": "mxfp8", | |
| "language_model.model.layers.3.mlp.switch_mlp.up_proj": "mxfp8", | |
| "language_model.model.layers.3.self_attn.k_proj": "mxfp8", | |
| "language_model.model.layers.3.self_attn.o_proj": "mxfp8", | |
| "language_model.model.layers.3.self_attn.q_proj": "mxfp8", | |
| "language_model.model.layers.3.self_attn.v_proj": "mxfp8", | |
| "language_model.model.layers.4.mlp.shared_experts.down_proj": "mxfp8", | |
| "language_model.model.layers.4.mlp.shared_experts.gate_proj": "mxfp8", | |
| "language_model.model.layers.4.mlp.shared_experts.up_proj": "mxfp8", | |
| "language_model.model.layers.4.mlp.switch_mlp.down_proj": "mxfp8", | |
| "language_model.model.layers.4.mlp.switch_mlp.gate_proj": "mxfp8", | |
| "language_model.model.layers.4.mlp.switch_mlp.up_proj": "mxfp8", | |
| "language_model.model.layers.4.self_attn.k_proj": "mxfp8", | |
| "language_model.model.layers.4.self_attn.o_proj": "mxfp8", | |
| "language_model.model.layers.4.self_attn.q_proj": "mxfp8", | |
| "language_model.model.layers.4.self_attn.v_proj": "mxfp8", | |
| "language_model.model.layers.5.mlp.shared_experts.down_proj": "mxfp8", | |
| "language_model.model.layers.5.mlp.shared_experts.gate_proj": "mxfp8", | |
| "language_model.model.layers.5.mlp.shared_experts.up_proj": "mxfp8", | |
| "language_model.model.layers.5.mlp.switch_mlp.down_proj": "mxfp8", | |
| "language_model.model.layers.5.mlp.switch_mlp.gate_proj": "mxfp8", | |
| "language_model.model.layers.5.mlp.switch_mlp.up_proj": "mxfp8", | |
| "language_model.model.layers.5.self_attn.k_proj": "mxfp8", | |
| "language_model.model.layers.5.self_attn.o_proj": "mxfp8", | |
| "language_model.model.layers.5.self_attn.q_proj": "mxfp8", | |
| "language_model.model.layers.5.self_attn.v_proj": "mxfp8", | |
| "language_model.model.layers.6.mlp.shared_experts.down_proj": "mxfp8", | |
| "language_model.model.layers.6.mlp.shared_experts.gate_proj": "mxfp8", | |
| "language_model.model.layers.6.mlp.shared_experts.up_proj": "mxfp8", | |
| "language_model.model.layers.6.mlp.switch_mlp.down_proj": "mxfp8", | |
| "language_model.model.layers.6.mlp.switch_mlp.gate_proj": "mxfp8", | |
| "language_model.model.layers.6.mlp.switch_mlp.up_proj": "mxfp8", | |
| "language_model.model.layers.6.self_attn.k_proj": "mxfp8", | |
| "language_model.model.layers.6.self_attn.o_proj": "mxfp8", | |
| "language_model.model.layers.6.self_attn.q_proj": "mxfp8", | |
| "language_model.model.layers.6.self_attn.v_proj": "mxfp8", | |
| "language_model.model.layers.7.mlp.shared_experts.down_proj": "mxfp8", | |
| "language_model.model.layers.7.mlp.shared_experts.gate_proj": "mxfp8", | |
| "language_model.model.layers.7.mlp.shared_experts.up_proj": "mxfp8", | |
| "language_model.model.layers.7.mlp.switch_mlp.down_proj": "mxfp8", | |
| "language_model.model.layers.7.mlp.switch_mlp.gate_proj": "mxfp8", | |
| "language_model.model.layers.7.mlp.switch_mlp.up_proj": "mxfp8", | |
| "language_model.model.layers.7.self_attn.k_proj": "mxfp8", | |
| "language_model.model.layers.7.self_attn.o_proj": "mxfp8", | |
| "language_model.model.layers.7.self_attn.q_proj": "mxfp8", | |
| "language_model.model.layers.7.self_attn.v_proj": "mxfp8", | |
| "language_model.model.layers.8.mlp.shared_experts.down_proj": "mxfp8", | |
| "language_model.model.layers.8.mlp.shared_experts.gate_proj": "mxfp8", | |
| "language_model.model.layers.8.mlp.shared_experts.up_proj": "mxfp8", | |
| "language_model.model.layers.8.mlp.switch_mlp.down_proj": "mxfp8", | |
| "language_model.model.layers.8.mlp.switch_mlp.gate_proj": "mxfp8", | |
| "language_model.model.layers.8.mlp.switch_mlp.up_proj": "mxfp8", | |
| "language_model.model.layers.8.self_attn.k_proj": "mxfp8", | |
| "language_model.model.layers.8.self_attn.o_proj": "mxfp8", | |
| "language_model.model.layers.8.self_attn.q_proj": "mxfp8", | |
| "language_model.model.layers.8.self_attn.v_proj": "mxfp8", | |
| "language_model.model.layers.9.mlp.shared_experts.down_proj": "mxfp8", | |
| "language_model.model.layers.9.mlp.shared_experts.gate_proj": "mxfp8", | |
| "language_model.model.layers.9.mlp.shared_experts.up_proj": "mxfp8", | |
| "language_model.model.layers.9.mlp.switch_mlp.down_proj": "mxfp8", | |
| "language_model.model.layers.9.mlp.switch_mlp.gate_proj": "mxfp8", | |
| "language_model.model.layers.9.mlp.switch_mlp.up_proj": "mxfp8", | |
| "language_model.model.layers.9.self_attn.k_proj": "mxfp8", | |
| "language_model.model.layers.9.self_attn.o_proj": "mxfp8", | |
| "language_model.model.layers.9.self_attn.q_proj": "mxfp8", | |
| "language_model.model.layers.9.self_attn.v_proj": "mxfp8" | |
| } | |
| } | |