license: apache-2.0
library_name: mlx
base_model: infly/Infinity-Parser2-Flash
tags:
- mlx
- mlx-vlm
- ocr
- document-parsing
- vision-language
- quantized
- 8-bit
pipeline_tag: image-text-to-text
language:
- en
Infinity-Parser2-Flash MLX 8-bit
This model was converted to MLX format from infly/Infinity-Parser2-Flash and quantized to 8-bit (group size 64) using mlx-vlm version 0.5.0. Refer to the original model card for more details on the model.
Use with mlx-vlm
pip install -U mlx-vlm
The model is RL-tuned for the canonical layout-extraction prompt below — using a different prompt may yield unexpected output:
PROMPT=$(cat <<'EOF'
- Extract layout information from the provided PDF image.
- For each layout element, output its bbox, category, and the text content within the bbox.
- Bbox format: [x1, y1, x2, y2].
- Allowed layout categories: ['header', 'title', 'text', 'figure', 'table', 'formula', 'figure_caption', 'table_caption', 'formula_caption', 'figure_footnote', 'table_footnote', 'page_footnote', 'footer'].
- Text extraction and formatting:
1) For 'figure', the text field must be an empty string.
2) For 'formula', format text as LaTeX.
3) For 'table', format text as HTML.
4) For all other categories (e.g., text, title), format text as Markdown.
- The output text must be exactly the original text from the image, with no translation or rewriting.
- Sort all layout elements in human reading order.
- Final output must be a single JSON object.
EOF
)
python -m mlx_vlm.generate \
--model BotResources/Infinity-Parser2-Flash-mlx-q8 \
--max-tokens 32768 --temperature 0.0 \
--prompt "$PROMPT" \
--image <path_to_image>
Quantization quality
The full-precision counterpart is published at BotResources/Infinity-Parser2-Flash-mlx-bf16.
In a BotResources internal benchmark of 50 pages from various PDFs (text, tables, formulas, scans), the BF16 build and this 8-bit build produced byte-identical outputs on all 50 pages at temperature=0, top_p=1. Token count, character count, and final text are strictly equal between the two builds.
On the same Apple M4 Max (128 GB unified memory) only the runtime differs:
| Build | On-disk | Peak RAM | Generation |
|---|---|---|---|
| BF16 | 4.43 GB | 5.4 GB | 101 tok/s |
| 8-bit (this build) | 2.48 GB | 3.7 GB | 167 tok/s |
This 8-bit build is ~65 % faster per token and uses ~33 % less peak RAM, with no measured quality loss for this use case.
License
Inherits the Apache-2.0 license from the base model infly/Infinity-Parser2-Flash. All credit for the underlying model goes to the inflyAI team.