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
| { | |
| "architectures": [ | |
| "UnlimitedOCRForCausalLM" | |
| ], | |
| "bos_token_id": 0, | |
| "candidate_resolutions": [ | |
| [ | |
| 1024, | |
| 1024 | |
| ] | |
| ], | |
| "eos_token_id": 1, | |
| "first_k_dense_replace": 1, | |
| "global_view_pos": "head", | |
| "hidden_size": 1280, | |
| "intermediate_size": 6848, | |
| "kv_lora_rank": null, | |
| "language_config": { | |
| "architectures": [ | |
| "DeepseekOCRForCausalLM" | |
| ], | |
| "bos_token_id": 0, | |
| "eos_token_id": 1, | |
| "first_k_dense_replace": 1, | |
| "hidden_size": 1280, | |
| "intermediate_size": 6848, | |
| "kv_lora_rank": null, | |
| "lm_head": true, | |
| "max_position_embeddings": 32768, | |
| "moe_intermediate_size": 896, | |
| "n_group": 1, | |
| "n_routed_experts": 64, | |
| "n_shared_experts": 2, | |
| "num_attention_heads": 10, | |
| "num_experts_per_tok": 6, | |
| "num_hidden_layers": 12, | |
| "num_key_value_heads": 10, | |
| "q_lora_rank": null, | |
| "qk_nope_head_dim": 0, | |
| "qk_rope_head_dim": 0, | |
| "rm_head": false, | |
| "topk_group": 1, | |
| "topk_method": "greedy", | |
| "torch_dtype": "bfloat16", | |
| "use_mla": false, | |
| "v_head_dim": 128, | |
| "vocab_size": 129280, | |
| "sliding_window_size": 128, | |
| "sliding_window": 128 | |
| }, | |
| "lm_head": true, | |
| "max_position_embeddings": 32768, | |
| "model_type": "unlimited-ocr", | |
| "moe_intermediate_size": 896, | |
| "n_group": 1, | |
| "n_routed_experts": 64, | |
| "n_shared_experts": 2, | |
| "num_attention_heads": 10, | |
| "num_experts_per_tok": 6, | |
| "num_hidden_layers": 12, | |
| "num_key_value_heads": 10, | |
| "projector_config": { | |
| "input_dim": 2048, | |
| "model_type": "mlp_projector", | |
| "n_embed": 1280, | |
| "projector_type": "linear" | |
| }, | |
| "q_lora_rank": null, | |
| "qk_nope_head_dim": 0, | |
| "qk_rope_head_dim": 0, | |
| "quantization": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8", | |
| "language_model.model.embed_tokens": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.0.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.0.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.0.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.0.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.0.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.0.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.0.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.lm_head": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "affine" | |
| } | |
| }, | |
| "quantization_config": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8", | |
| "language_model.model.embed_tokens": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.0.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.0.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.0.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.0.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.0.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.0.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.0.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.1.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.2.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.3.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.4.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.5.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.6.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.7.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.8.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.9.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.10.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.self_attn.q_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.self_attn.k_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.self_attn.v_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.self_attn.o_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.mlp.switch_mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.mlp.switch_mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.mlp.switch_mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.mlp.shared_experts.gate_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.mlp.shared_experts.up_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.model.layers.11.mlp.shared_experts.down_proj": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "mxfp8" | |
| }, | |
| "language_model.lm_head": { | |
| "group_size": 32, | |
| "bits": 8, | |
| "mode": "affine" | |
| } | |
| }, | |
| "rm_head": false, | |
| "sliding_window": 128, | |
| "sliding_window_size": 128, | |
| "tile_tag": "2D", | |
| "topk_group": 1, | |
| "topk_method": "greedy", | |
| "transformers_version": "4.46.3", | |
| "use_mla": false, | |
| "v_head_dim": 128, | |
| "vision_config": { | |
| "image_size": 1024, | |
| "mlp_ratio": 3.7362, | |
| "model_name": "deeplip_b_l", | |
| "model_type": "vision", | |
| "width": { | |
| "clip-l-14-224": { | |
| "heads": 16, | |
| "image_size": 224, | |
| "layers": 24, | |
| "patch_size": 14, | |
| "width": 1024 | |
| }, | |
| "sam_vit_b": { | |
| "downsample_channels": [ | |
| 512, | |
| 1024 | |
| ], | |
| "global_attn_indexes": [ | |
| 2, | |
| 5, | |
| 8, | |
| 11 | |
| ], | |
| "heads": 12, | |
| "layers": 12, | |
| "width": 768 | |
| } | |
| } | |
| }, | |
| "vocab_size": 129280 | |
| } | |