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
File size: 15,572 Bytes
fd0c04f 6708e4d fd0c04f 6708e4d fd0c04f 6708e4d fd0c04f 6708e4d fd0c04f 6708e4d fd0c04f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 | {
"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"
}
}
|