dots.mocr-W4A16-AutoRound

dots.mocr W4A16 AutoRound Quantized

This repository provides production-grade W4A16 quantized weights for dots-studio/dots.mocr using Intel AutoRound.

Available Variants

• Vishva007/dots.mocr-W4A16-AutoRound: Native AutoRound format.
• Vishva007/dots.mocr-W4A16-AutoRound-GPTQ: Exported GPTQ format optimized for direct serving in vLLM.

---

Quantization Details

The quantization recipe was tuned for high accuracy and long-context multimodal parsing while preventing degradation of visual features:

• Bits / Weight Type: 4-bit weights, 16-bit activations (W4A16)
• Group Size: 32 (Symmetric: True)
• Vision Tower Preservation: quant_nontext_module=False (vision encoder preserved in full precision to retain document grounding accuracy)
• Tuning Recipe: 1,500 iterations, 1,024 samples, sequence length 8,192 with torch.compile acceleration.

---

Deployment with vLLM

[!IMPORTANT] KV Cache Precision Warning: Always keep --kv-cache-dtype set to auto or bfloat16. Do not use low-precision KV cache formats (such as fp8), as doing so will cause the model to fail to generate any output or return empty responses. For high-throughput document parsing, serve the AutoRound variant directly with vLLM:

vllm serve Vishva007/dots.mocr-W4A16-AutoRound \
  --host 0.0.0.0 \
  --port 8000 \
  --trust-remote-code \
  --chat-template-content-format string \
  --dtype bfloat16 \
  --kv-cache-dtype auto \
  --max-model-len 32768 \
  --max-num-seqs 128 \
  --gpu-memory-utilization 0.90 \
  --enable-prefix-caching \
  --enable-chunked-prefill

Benchmark Highlights (Original Base Model)

• olmOCR-Bench: 83.9 Overall
• OmniDocBench (v1.5): 0.031 TextEdit / 0.029 Read OrderEdit
• Native Structured Output: Direct image-to-Markdown, HTML tables, LaTeX formulas, and SVG generation.

---

🚀 Deploy on RunPod

One-click launch environments pre-configured with PyTorch, CUDA, and dependencies for fine-tuning or quantization.

🎁 Need GPU compute? Sign up via RunPod and get $5–$500 in free credits when you add your first $10.

PyTorch 2.14

PyTorch 2.13

PyTorch 2.12

---

Acknowledgements

• Base model developed by dots-studio.
• Quantized using Intel AutoRound.