Skip to content

Latest commit

 

History

History
58 lines (46 loc) · 1.53 KB

File metadata and controls

58 lines (46 loc) · 1.53 KB
model_id mineru2.5
backend vllm
hardware
gpu vram_min_gb
AMD gfx1100
48
environment
type rocm
docker
7.2
command python -m mineru_rocm runner \ --backend vlm-vllm --platform linux-rocm \ --server-url http://127.0.0.1:8265/v1 --api-model-name mineru-pro \ --img-dir /root/datasets/OmniDocBench_data/images \ --out-dir /tmp/mineru-predictions omnidocbench-rocm run --stage score --platform linux-rocm --cdm \ --predictions-dir /tmp/mineru-predictions \ --run-stats /tmp/mineru-predictions/_run_stats.json --version v16
expected_overall
value tolerance
95.56
0.5

Reproduce MinerU2.5-Pro VLM 95.56 on AMD ROCm

Prerequisites

rocminfo | grep -E "Name:|VRAM"    # must show gfx1100 + ≥48 GB
ls -la /dev/kfd                     # must exist

Quickstart

# Start mineru-pro server
python -m mineru_rocm serve-vlm --port 8265

# Run adapter
python -m mineru_rocm runner --backend vlm-vllm --platform linux-rocm \
  --server-url http://127.0.0.1:8265/v1 --api-model-name mineru-pro \
  --img-dir /root/datasets/OmniDocBench_data/images \
  --out-dir /tmp/mineru-predictions

# Score
omnidocbench-rocm run --stage score --platform linux-rocm --cdm \
  --predictions-dir /tmp/mineru-predictions \
  --run-stats /tmp/mineru-predictions/_run_stats.json --version v16

Expected output

Overall 95.56 (±0.5). Text 96.41, Table TEDS 93.54%, Formula CDM 96.73%.

If it fails

See OmniDocBench-ROCm pitfalls.