infinity-parser2 0.1.0__tar.gz → 0.2.0__tar.gz

{infinity_parser2-0.1.0 → infinity_parser2-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: infinity_parser2
-Version: 0.1.0
+Version: 0.2.0
 Summary: Document parsing Python package supporting PDF and image parsing using Infinity-Parser2-Pro model.
 Home-page: https://github.com/infly-ai/INF-MLLM
 Author: INF Tech
@@ -53,7 +53,33 @@ Dynamic: summary
 
 # Infinity-Parser2
 
-Infinity-Parser2 is a document parsing tool powered by the Infinity-Parser2-Pro model. It converts **PDF files** and **images** (PNG, JPG, WEBP) into structured Markdown or JSON with layout information.
+<p align="center">
+    <img src="https://raw.githubusercontent.com/infly-ai/INF-MLLM/main/Infinity-Parser2/assets/logo.png" width="400"/>
+<p>
+
+<p align="center">
+🤗 <a href="https://huggingface.co/infly/Infinity-Parser2-Pro">Model</a> |
+📊 <a>Dataset (coming soon...)</a> |
+📄 <a>Paper (coming soon...)</a> |
+🚀 <a>Demo (coming soon...)</a>
+</p>
+
+## Introduction
+
+We are excited to release Infinity-Parser2-Pro, our latest flagship document understanding model that achieves a new state-of-the-art on olmOCR-Bench with a score of 86.7%, surpassing frontier models such as DeepSeek-OCR-2, PaddleOCR-VL, and dots.mocr. Building on our previous model Infinity-Parser-7B, we have significantly enhanced our data engine and multi-task reinforcement learning approach. This enables the model to consolidate robust multi-modal parsing capabilities into a unified architecture, delivering brand-new zero-shot capabilities for diverse real-world business scenarios.
+
+### Key Features
+
+- **Upgraded Data Engine**: We have comprehensively enhanced our synthetic data engine to support both fixed-layout and flexible-layout document formats. By generating over 1 million diverse full-text samples covering a wide range of document layouts, combined with a dynamic adaptive sampling strategy, we ensure highly balanced and robust multi-task learning across various document types.
+- **Multi-Task Reinforcement Learning**: We designed a novel verifiable reward system to support Joint Reinforcement Learning (RL), enabling seamless and simultaneous co-optimization of multiple complex tasks, including doc2json and doc2markdown.
+- **Breakthrough Parsing Performance**: It substantially outperforms our previous 7B model, achieving 86.7% on olmOCR-Bench, surpassing frontier models such as DeepSeek-OCR-2, PaddleOCR-VL, and dots.mocr.
+- **Inference Acceleration**: By adopting the highly efficient MoE architecture, our inference throughput has increased by 21% (from 441 to 534 tokens/sec), reducing deployment latency and costs.
+
+## Performance
+
+<p align="left">
+    <img src="https://raw.githubusercontent.com/infly-ai/INF-MLLM/main/Infinity-Parser2/assets/document_parsing_performance_evaluation.png" width="1200"/>
+<p>
 
 ## Quick Start
 
@@ -62,13 +88,17 @@ Infinity-Parser2 is a document parsing tool powered by the Infinity-Parser2-Pro
 #### Pre-requisites
 
 ```bash
-# Install PyTorch (CUDA). Find the proper version on the [official site](https://pytorch.org/get-started/previous-versions) based on your CUDA version.
+# Create a Conda environment (Optional)
+conda create -n infinity_parser2 python=3.12
+conda activate infinity_parser2
+
+# Install PyTorch (CUDA). Find the proper version at https://pytorch.org/get-started/previous-versions based on your CUDA version.
 pip install torch==2.10.0 torchvision==0.25.0 torchaudio==2.10.0 --index-url https://download.pytorch.org/whl/cu128
 
 # Install FlashAttention (required for NVIDIA GPUs).
 # This command builds flash-attn from source, which can take 10 to 30 minutes.
 pip install flash-attn==2.8.3 --no-build-isolation
-# For Hopper GPUs (e.g. H100, H800), we recommend FlashAttention-3 instead. See the [official guide](https://github.com/Dao-AILab/flash-attention).
+# For Hopper GPUs (e.g. H100, H800), we recommend FlashAttention-3 instead. See the official guide at https://github.com/Dao-AILab/flash-attention.
 
 # Install vLLM
 # NOTE: you may need to run the command below to resolve triton and numpy conflicts before installing vllm.
@@ -95,6 +125,8 @@ pip install -e .
 The `parser` command is the fastest way to get started.
 
 ```bash
+# NOTE: The Infinity-Parser2 model will be automatically downloaded on the first run.
+
 # Parse a PDF (outputs Markdown by default)
 parser demo_data/demo.pdf
 
@@ -122,6 +154,8 @@ parser --help
 #### Python API
 
 ```python
+# NOTE: The Infinity-Parser2 model will be automatically downloaded on the first run.
+
 from infinity_parser2 import InfinityParser2
 
 parser = InfinityParser2()
@@ -154,7 +188,7 @@ result = parser.parse("demo_data/demo.pdf", task_type="doc2md")
 
 # Custom prompt
 result = parser.parse("demo_data/demo.pdf", task_type="custom",
-                      custom_prompt="Extract the title and authors only.")
+                      custom_prompt="Please transform the document's contents into Markdown format.")
 
 # Batch processing with custom batch size
 result = parser.parse("demo_data", batch_size=8)
@@ -308,3 +342,7 @@ print(cache.resolve_model_path("infly/Infinity-Parser2-Pro"))
 - Python 3.12+
 - CUDA-compatible GPU
 - See `setup.py` for full dependency list.
+
+## Acknowledgments
+
+We would like to thank [Qwen3.5](https://github.com/QwenLM/Qwen3.5), [ms-swift](https://github.com/modelscope/ms-swift), [VeRL](https://github.com/verl-project/verl), [lmms-eval](https://github.com/EvolvingLMMs-Lab/lmms-eval), [olmocr](https://huggingface.co/datasets/allenai/olmOCR-bench), [PaddleOCR-VL](https://github.com/PaddlePaddle/PaddleOCR), [MinerU](https://github.com/opendatalab/MinerU), [dots.ocr](https://github.com/rednote-hilab/dots.ocr), [Chandra-OCR-2](https://github.com/datalab-to/chandra) for providing dataset, code and models.

{infinity_parser2-0.1.0 → infinity_parser2-0.2.0}/README.md

@@ -1,6 +1,32 @@
 # Infinity-Parser2
 
-Infinity-Parser2 is a document parsing tool powered by the Infinity-Parser2-Pro model. It converts **PDF files** and **images** (PNG, JPG, WEBP) into structured Markdown or JSON with layout information.
+<p align="center">
+    <img src="https://raw.githubusercontent.com/infly-ai/INF-MLLM/main/Infinity-Parser2/assets/logo.png" width="400"/>
+<p>
+
+<p align="center">
+🤗 <a href="https://huggingface.co/infly/Infinity-Parser2-Pro">Model</a> |
+📊 <a>Dataset (coming soon...)</a> |
+📄 <a>Paper (coming soon...)</a> |
+🚀 <a>Demo (coming soon...)</a>
+</p>
+
+## Introduction
+
+We are excited to release Infinity-Parser2-Pro, our latest flagship document understanding model that achieves a new state-of-the-art on olmOCR-Bench with a score of 86.7%, surpassing frontier models such as DeepSeek-OCR-2, PaddleOCR-VL, and dots.mocr. Building on our previous model Infinity-Parser-7B, we have significantly enhanced our data engine and multi-task reinforcement learning approach. This enables the model to consolidate robust multi-modal parsing capabilities into a unified architecture, delivering brand-new zero-shot capabilities for diverse real-world business scenarios.
+
+### Key Features
+
+- **Upgraded Data Engine**: We have comprehensively enhanced our synthetic data engine to support both fixed-layout and flexible-layout document formats. By generating over 1 million diverse full-text samples covering a wide range of document layouts, combined with a dynamic adaptive sampling strategy, we ensure highly balanced and robust multi-task learning across various document types.
+- **Multi-Task Reinforcement Learning**: We designed a novel verifiable reward system to support Joint Reinforcement Learning (RL), enabling seamless and simultaneous co-optimization of multiple complex tasks, including doc2json and doc2markdown.
+- **Breakthrough Parsing Performance**: It substantially outperforms our previous 7B model, achieving 86.7% on olmOCR-Bench, surpassing frontier models such as DeepSeek-OCR-2, PaddleOCR-VL, and dots.mocr.
+- **Inference Acceleration**: By adopting the highly efficient MoE architecture, our inference throughput has increased by 21% (from 441 to 534 tokens/sec), reducing deployment latency and costs.
+
+## Performance
+
+<p align="left">
+    <img src="https://raw.githubusercontent.com/infly-ai/INF-MLLM/main/Infinity-Parser2/assets/document_parsing_performance_evaluation.png" width="1200"/>
+<p>
 
 ## Quick Start
 
@@ -9,13 +35,17 @@ Infinity-Parser2 is a document parsing tool powered by the Infinity-Parser2-Pro
 #### Pre-requisites
 
 ```bash
-# Install PyTorch (CUDA). Find the proper version on the [official site](https://pytorch.org/get-started/previous-versions) based on your CUDA version.
+# Create a Conda environment (Optional)
+conda create -n infinity_parser2 python=3.12
+conda activate infinity_parser2
+
+# Install PyTorch (CUDA). Find the proper version at https://pytorch.org/get-started/previous-versions based on your CUDA version.
 pip install torch==2.10.0 torchvision==0.25.0 torchaudio==2.10.0 --index-url https://download.pytorch.org/whl/cu128
 
 # Install FlashAttention (required for NVIDIA GPUs).
 # This command builds flash-attn from source, which can take 10 to 30 minutes.
 pip install flash-attn==2.8.3 --no-build-isolation
-# For Hopper GPUs (e.g. H100, H800), we recommend FlashAttention-3 instead. See the [official guide](https://github.com/Dao-AILab/flash-attention).
+# For Hopper GPUs (e.g. H100, H800), we recommend FlashAttention-3 instead. See the official guide at https://github.com/Dao-AILab/flash-attention.
 
 # Install vLLM
 # NOTE: you may need to run the command below to resolve triton and numpy conflicts before installing vllm.
@@ -42,6 +72,8 @@ pip install -e .
 The `parser` command is the fastest way to get started.
 
 ```bash
+# NOTE: The Infinity-Parser2 model will be automatically downloaded on the first run.
+
 # Parse a PDF (outputs Markdown by default)
 parser demo_data/demo.pdf
 
@@ -69,6 +101,8 @@ parser --help
 #### Python API
 
 ```python
+# NOTE: The Infinity-Parser2 model will be automatically downloaded on the first run.
+
 from infinity_parser2 import InfinityParser2
 
 parser = InfinityParser2()
@@ -101,7 +135,7 @@ result = parser.parse("demo_data/demo.pdf", task_type="doc2md")
 
 # Custom prompt
 result = parser.parse("demo_data/demo.pdf", task_type="custom",
-                      custom_prompt="Extract the title and authors only.")
+                      custom_prompt="Please transform the document's contents into Markdown format.")
 
 # Batch processing with custom batch size
 result = parser.parse("demo_data", batch_size=8)
@@ -255,3 +289,7 @@ print(cache.resolve_model_path("infly/Infinity-Parser2-Pro"))
 - Python 3.12+
 - CUDA-compatible GPU
 - See `setup.py` for full dependency list.
+
+## Acknowledgments
+
+We would like to thank [Qwen3.5](https://github.com/QwenLM/Qwen3.5), [ms-swift](https://github.com/modelscope/ms-swift), [VeRL](https://github.com/verl-project/verl), [lmms-eval](https://github.com/EvolvingLMMs-Lab/lmms-eval), [olmocr](https://huggingface.co/datasets/allenai/olmOCR-bench), [PaddleOCR-VL](https://github.com/PaddlePaddle/PaddleOCR), [MinerU](https://github.com/opendatalab/MinerU), [dots.ocr](https://github.com/rednote-hilab/dots.ocr), [Chandra-OCR-2](https://github.com/datalab-to/chandra) for providing dataset, code and models.

{infinity_parser2-0.1.0 → infinity_parser2-0.2.0}/infinity_parser2/__init__.py

@@ -1,6 +1,6 @@
 """Infinity-Parser2: Document parsing Python package."""
 
-__version__ = "0.1.0"
+__version__ = "0.2.0"
 
 from .parser import InfinityParser2
 from .backends import (

{infinity_parser2-0.1.0 → infinity_parser2-0.2.0}/infinity_parser2/cli.py

@@ -26,28 +26,28 @@ def build_parser() -> argparse.ArgumentParser:
         epilog="""
 Examples:
   # Parse a PDF file (default: doc2json -> markdown output)
-  parser document.pdf
+  parser demo_data/demo.pdf
 
   # Parse with doc2md task type
-  parser document.pdf --task doc2md
+  parser demo_data/demo.pdf --task doc2md
 
   # Parse with custom prompt
-  parser document.pdf --task custom --prompt "Extract the title and authors"
+  parser demo_data/demo.pdf --task custom --prompt "Please transform the document's contents into Markdown format."
 
   # Parse multiple files
-  parser doc1.pdf doc2.png --output-dir ./results
+  parser demo_data/demo.pdf demo_data/demo.png --output-dir ./results
 
   # Parse a directory
-  parser ./docs --output-dir ./results
+  parser demo_data --output-dir ./results
 
   # Output raw JSON
-  parser document.pdf --output-format json
+  parser demo_data/demo.pdf --output-format json
 
   # Use transformers backend
-  parser document.pdf --backend transformers
+  parser demo_data/demo.pdf --backend transformers
 
   # Use vllm-server backend
-  parser document.pdf --backend vllm-server --api-url http://localhost:8000/v1/chat/completions
+  parser demo_data/demo.pdf --backend vllm-server --api-url http://localhost:8000/v1/chat/completions
         """,
     )
 
@@ -136,7 +136,7 @@ Examples:
     parser.add_argument(
         "--version",
         action="version",
-        version="Infinity-Parser2 0.1.0",
+        version="Infinity-Parser2 0.2.0",
     )
 
     return parser

{infinity_parser2-0.1.0 → infinity_parser2-0.2.0}/infinity_parser2/parser.py

@@ -52,7 +52,7 @@ class InfinityParser2:
     Example:
         >>> from infinity_parser2 import InfinityParser2
         >>> parser = InfinityParser2(model_name="infly/Infinity-Parser2-Pro")
-        >>> result = parser.parse("document.pdf")
+        >>> result = parser.parse("demo_data/demo.pdf")
     """
 
     def __init__(
@@ -183,13 +183,13 @@ class InfinityParser2:
         Example:
             >>> parser = InfinityParser2()
             >>> # Single file, returns str
-            >>> result = parser.parse("document.pdf")
+            >>> result = parser.parse("demo_data/demo.pdf")
             >>> # Multiple files, returns List[str]
-            >>> result = parser.parse(["doc1.pdf", "doc2.pdf"])
+            >>> result = parser.parse(["demo_data/demo.pdf", "demo_data/demo.png"])
             >>> # Directory, returns Dict[str, str]
-            >>> result = parser.parse("/path/to/docs")
+            >>> result = parser.parse("./demo_data")
             >>> # Save results to output_dir, returns None
-            >>> parser.parse("document.pdf", output_dir="./output")
+            >>> parser.parse("demo_data/demo.pdf", output_dir="./output")
         """
         if task_type not in SUPPORTED_TASK_TYPES:
             raise ValueError(f"task_type must be one of {SUPPORTED_TASK_TYPES}, got '{task_type}'")
@@ -204,6 +204,7 @@ class InfinityParser2:
             )
 
         prompt = self._resolve_prompt(task_type, custom_prompt)
+        print(f"[Infinity-Parser2] task_type: {task_type}, prompt: {prompt}")
 
         is_directory = isinstance(input_data, str) and os.path.isdir(input_data)
         file_paths = normalize_input(input_data)

{infinity_parser2-0.1.0 → infinity_parser2-0.2.0}/infinity_parser2.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: infinity_parser2
-Version: 0.1.0
+Version: 0.2.0
 Summary: Document parsing Python package supporting PDF and image parsing using Infinity-Parser2-Pro model.
 Home-page: https://github.com/infly-ai/INF-MLLM
 Author: INF Tech
@@ -53,7 +53,33 @@ Dynamic: summary
 
 # Infinity-Parser2
 
-Infinity-Parser2 is a document parsing tool powered by the Infinity-Parser2-Pro model. It converts **PDF files** and **images** (PNG, JPG, WEBP) into structured Markdown or JSON with layout information.
+<p align="center">
+    <img src="https://raw.githubusercontent.com/infly-ai/INF-MLLM/main/Infinity-Parser2/assets/logo.png" width="400"/>
+<p>
+
+<p align="center">
+🤗 <a href="https://huggingface.co/infly/Infinity-Parser2-Pro">Model</a> |
+📊 <a>Dataset (coming soon...)</a> |
+📄 <a>Paper (coming soon...)</a> |
+🚀 <a>Demo (coming soon...)</a>
+</p>
+
+## Introduction
+
+We are excited to release Infinity-Parser2-Pro, our latest flagship document understanding model that achieves a new state-of-the-art on olmOCR-Bench with a score of 86.7%, surpassing frontier models such as DeepSeek-OCR-2, PaddleOCR-VL, and dots.mocr. Building on our previous model Infinity-Parser-7B, we have significantly enhanced our data engine and multi-task reinforcement learning approach. This enables the model to consolidate robust multi-modal parsing capabilities into a unified architecture, delivering brand-new zero-shot capabilities for diverse real-world business scenarios.
+
+### Key Features
+
+- **Upgraded Data Engine**: We have comprehensively enhanced our synthetic data engine to support both fixed-layout and flexible-layout document formats. By generating over 1 million diverse full-text samples covering a wide range of document layouts, combined with a dynamic adaptive sampling strategy, we ensure highly balanced and robust multi-task learning across various document types.
+- **Multi-Task Reinforcement Learning**: We designed a novel verifiable reward system to support Joint Reinforcement Learning (RL), enabling seamless and simultaneous co-optimization of multiple complex tasks, including doc2json and doc2markdown.
+- **Breakthrough Parsing Performance**: It substantially outperforms our previous 7B model, achieving 86.7% on olmOCR-Bench, surpassing frontier models such as DeepSeek-OCR-2, PaddleOCR-VL, and dots.mocr.
+- **Inference Acceleration**: By adopting the highly efficient MoE architecture, our inference throughput has increased by 21% (from 441 to 534 tokens/sec), reducing deployment latency and costs.
+
+## Performance
+
+<p align="left">
+    <img src="https://raw.githubusercontent.com/infly-ai/INF-MLLM/main/Infinity-Parser2/assets/document_parsing_performance_evaluation.png" width="1200"/>
+<p>
 
 ## Quick Start
 
@@ -62,13 +88,17 @@ Infinity-Parser2 is a document parsing tool powered by the Infinity-Parser2-Pro
 #### Pre-requisites
 
 ```bash
-# Install PyTorch (CUDA). Find the proper version on the [official site](https://pytorch.org/get-started/previous-versions) based on your CUDA version.
+# Create a Conda environment (Optional)
+conda create -n infinity_parser2 python=3.12
+conda activate infinity_parser2
+
+# Install PyTorch (CUDA). Find the proper version at https://pytorch.org/get-started/previous-versions based on your CUDA version.
 pip install torch==2.10.0 torchvision==0.25.0 torchaudio==2.10.0 --index-url https://download.pytorch.org/whl/cu128
 
 # Install FlashAttention (required for NVIDIA GPUs).
 # This command builds flash-attn from source, which can take 10 to 30 minutes.
 pip install flash-attn==2.8.3 --no-build-isolation
-# For Hopper GPUs (e.g. H100, H800), we recommend FlashAttention-3 instead. See the [official guide](https://github.com/Dao-AILab/flash-attention).
+# For Hopper GPUs (e.g. H100, H800), we recommend FlashAttention-3 instead. See the official guide at https://github.com/Dao-AILab/flash-attention.
 
 # Install vLLM
 # NOTE: you may need to run the command below to resolve triton and numpy conflicts before installing vllm.
@@ -95,6 +125,8 @@ pip install -e .
 The `parser` command is the fastest way to get started.
 
 ```bash
+# NOTE: The Infinity-Parser2 model will be automatically downloaded on the first run.
+
 # Parse a PDF (outputs Markdown by default)
 parser demo_data/demo.pdf
 
@@ -122,6 +154,8 @@ parser --help
 #### Python API
 
 ```python
+# NOTE: The Infinity-Parser2 model will be automatically downloaded on the first run.
+
 from infinity_parser2 import InfinityParser2
 
 parser = InfinityParser2()
@@ -154,7 +188,7 @@ result = parser.parse("demo_data/demo.pdf", task_type="doc2md")
 
 # Custom prompt
 result = parser.parse("demo_data/demo.pdf", task_type="custom",
-                      custom_prompt="Extract the title and authors only.")
+                      custom_prompt="Please transform the document's contents into Markdown format.")
 
 # Batch processing with custom batch size
 result = parser.parse("demo_data", batch_size=8)
@@ -308,3 +342,7 @@ print(cache.resolve_model_path("infly/Infinity-Parser2-Pro"))
 - Python 3.12+
 - CUDA-compatible GPU
 - See `setup.py` for full dependency list.
+
+## Acknowledgments
+
+We would like to thank [Qwen3.5](https://github.com/QwenLM/Qwen3.5), [ms-swift](https://github.com/modelscope/ms-swift), [VeRL](https://github.com/verl-project/verl), [lmms-eval](https://github.com/EvolvingLMMs-Lab/lmms-eval), [olmocr](https://huggingface.co/datasets/allenai/olmOCR-bench), [PaddleOCR-VL](https://github.com/PaddlePaddle/PaddleOCR), [MinerU](https://github.com/opendatalab/MinerU), [dots.ocr](https://github.com/rednote-hilab/dots.ocr), [Chandra-OCR-2](https://github.com/datalab-to/chandra) for providing dataset, code and models.

{infinity_parser2-0.1.0 → infinity_parser2-0.2.0}/setup.py

@@ -32,7 +32,7 @@ install_requires = [
 
 setup(
     name="infinity_parser2",
-    version="0.1.0",
+    version="0.2.0",
     description="Document parsing Python package supporting PDF and image parsing using Infinity-Parser2-Pro model.",
     long_description=open("README.md", "r", encoding="utf-8").read(),
     long_description_content_type="text/markdown",