Unlimited-OCR (MLX 8bit, verified)

Baiduの Unlimited-OCR(3.3B・DeepSeek-V2系MoE+MLA+deepencoder構成のドキュメントパースモデル)を、公式重みからMLX 8bit(group size 64、実効9.41bpw)に量子化したものです。日本語ビジネス文書での検証済みで公開しています。

ページ画像から、レイアウト検出ボックス(<|det|>種別 [x1,y1,x2,y2]<|/det|>)付きで読み順どおりのテキスト(表はHTML)を出力します。

なぜ公開するか

公開時点で存在した本モデルの既存MLX変換を同一プロトコルで検証したところ、現行mlx-vlmでロードできないもの(model_type改名)や、トークナイザ不整合により出力がbyte-BPE片のまま文字化けするものを確認しました。本リポジトリは公式重みから変換し、下記の検証を通した上で公開しています。

変換品質の検証(tokimoa)

日本語ビジネス文書5種(請求書・議事録・契約条項・チェックシート・英日混在技術文書、正解文字列59項目)で検証:

  • テキスト内容がbf16版と5/5画像で完全一致(差分は検出ボックス座標の±数pxのみ)
  • 正解文字列カバレッジ 57/59(bf16版と同一。差分2件は記号置換のみ)
  • 同一入力の反復で出力完全一致(決定論的)

実測速度(M4 Max 36GB)

  • 生成: 312 tok/s / ピークメモリ: 5.9GB(MoEのアクティブパラメータが小さいため高速)

使い方

チャットテンプレートではなく、素のプロンプトに <image> トークンを含める形式です:

# pip install mlx-vlm  (0.6.7で動作確認)
from mlx_vlm import load, generate

model, processor = load("tokimoa/unlimited-ocr-mlx-8bit")
out = generate(model, processor, "<image>document parsing.",
               image=["page.png"], max_tokens=4096, temperature=0.0)
print(out.text)

複数ページは "<image>Multi page parsing."(上流README準拠)。

変換情報

  • 変換元: baidu/Unlimited-OCR(公式重み)
  • 変換: mlx-vlm 0.6.7 convert -q --q-bits 8(3.8GB)
  • 変換者: tokimoa

ライセンス

MIT(元モデルに準拠)

Downloads last month
61
Safetensors
Model size
1B params
Tensor type
BF16
·
U32
·
MLX
Hardware compatibility
Log In to add your hardware

8-bit

Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for tokimoa/unlimited-ocr-mlx-8bit

Quantized
(27)
this model