mirror of
https://github.com/DS4SD/docling.git
synced 2025-12-08 20:58:11 +00:00
47 lines
1.1 KiB
Python
Vendored
47 lines
1.1 KiB
Python
Vendored
from docling.datamodel import vlm_model_specs
|
|
from docling.datamodel.base_models import InputFormat
|
|
from docling.datamodel.pipeline_options import (
|
|
VlmPipelineOptions,
|
|
)
|
|
from docling.document_converter import DocumentConverter, PdfFormatOption
|
|
from docling.pipeline.vlm_pipeline import VlmPipeline
|
|
|
|
source = "https://arxiv.org/pdf/2501.17887"
|
|
|
|
###### USING SIMPLE DEFAULT VALUES
|
|
# - GraniteDocling model
|
|
# - Using the transformers framework
|
|
|
|
converter = DocumentConverter(
|
|
format_options={
|
|
InputFormat.PDF: PdfFormatOption(
|
|
pipeline_cls=VlmPipeline,
|
|
),
|
|
}
|
|
)
|
|
|
|
doc = converter.convert(source=source).document
|
|
|
|
print(doc.export_to_markdown())
|
|
|
|
|
|
###### USING MACOS MPS ACCELERATOR
|
|
# For more options see the compare_vlm_models.py example.
|
|
|
|
pipeline_options = VlmPipelineOptions(
|
|
vlm_options=vlm_model_specs.GRANITEDOCLING_MLX,
|
|
)
|
|
|
|
converter = DocumentConverter(
|
|
format_options={
|
|
InputFormat.PDF: PdfFormatOption(
|
|
pipeline_cls=VlmPipeline,
|
|
pipeline_options=pipeline_options,
|
|
),
|
|
}
|
|
)
|
|
|
|
doc = converter.convert(source=source).document
|
|
|
|
print(doc.export_to_markdown())
|