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
| { | |
| "quantization_method": "mxfp8", | |
| "description": "Quantization layout of the reference sahilchachra checkpoint used by the default upload workflow. MXFP8 is applied to quantizable language modules, token embeddings, the LM head, and the projector; the vision encoders and non-quantizable normalization/routing weights remain in bfloat16.", | |
| "base_model": "baidu/Unlimited-OCR", | |
| "reference_quantization": "sahilchachra/unlimited-ocr-mxfp8-mlx", | |
| "quantized_components": { | |
| "language_model.linear": "mxfp8", | |
| "language_model.moe.experts": "mxfp8", | |
| "language_model.token_embeddings": "mxfp8", | |
| "language_model.lm_head": "mxfp8", | |
| "vision_projector": "mxfp8" | |
| }, | |
| "preserved_bf16_components": { | |
| "vision_encoder": "bfloat16", | |
| "normalization_layers": "bfloat16", | |
| "moe_routing_gates": "bfloat16" | |
| }, | |
| "effective_bits_per_weight": 9.19, | |
| "model_size_gb": 3.83, | |
| "conversion_tool": "mlx-vlm quantizers", | |
| "notes": [ | |
| "This file describes the reference weights copied by scripts/upload_model.py; that workflow does not run a new conversion", | |
| "The safetensors index contains MXFP8 scale tensors for the token embeddings, LM head, and projector, but not the vision encoders", | |
| "quantization/mixed_precision_convert.py can produce a different OCR-aware layout from the BF16 base model", | |
| "Published config.json uses model_type 'unlimited-ocr' to select mlx-vlm's native R-SWA implementation" | |
| ] | |
| } | |