From 3ff1712787471aa638fe81e762f9df195f3bbf4f Mon Sep 17 00:00:00 2001 From: Michele Dolfi Date: Sun, 1 Jun 2025 18:29:20 +0200 Subject: [PATCH] rename pipeline_vlm_model_spec Signed-off-by: Michele Dolfi --- docling/cli/main.py | 4 ++-- docling/datamodel/pipeline_options.py | 2 +- .../{pipeline_vlm_model_spec.py => vlm_model_spec.py} | 0 docling/utils/model_downloader.py | 4 ++-- docs/examples/compare_vlm_models.py | 2 +- docs/examples/minimal_vlm_pipeline.py | 2 +- 6 files changed, 7 insertions(+), 7 deletions(-) rename docling/datamodel/{pipeline_vlm_model_spec.py => vlm_model_spec.py} (100%) diff --git a/docling/cli/main.py b/docling/cli/main.py index 88edde0b..4296128c 100644 --- a/docling/cli/main.py +++ b/docling/cli/main.py @@ -41,14 +41,14 @@ from docling.datamodel.pipeline_options import ( TableFormerMode, VlmPipelineOptions, ) -from docling.datamodel.pipeline_vlm_model_spec import ( +from docling.datamodel.settings import settings +from docling.datamodel.vlm_model_spec import ( GRANITE_VISION_OLLAMA, GRANITE_VISION_TRANSFORMERS, SMOLDOCLING_MLX, SMOLDOCLING_TRANSFORMERS, VlmModelType, ) -from docling.datamodel.settings import settings from docling.document_converter import DocumentConverter, FormatOption, PdfFormatOption from docling.models.factories import get_ocr_factory from docling.pipeline.vlm_pipeline import VlmPipeline diff --git a/docling/datamodel/pipeline_options.py b/docling/datamodel/pipeline_options.py index 4c73a5b1..5b870550 100644 --- a/docling/datamodel/pipeline_options.py +++ b/docling/datamodel/pipeline_options.py @@ -23,7 +23,7 @@ from docling.datamodel.pipeline_options_vlm_model import ( InferenceFramework, ResponseFormat, ) -from docling.datamodel.pipeline_vlm_model_spec import ( +from docling.datamodel.vlm_model_spec import ( GRANITE_VISION_OLLAMA as granite_vision_vlm_ollama_conversion_options, GRANITE_VISION_TRANSFORMERS as granite_vision_vlm_conversion_options, SMOLDOCLING_MLX as smoldocling_vlm_mlx_conversion_options, diff --git a/docling/datamodel/pipeline_vlm_model_spec.py b/docling/datamodel/vlm_model_spec.py similarity index 100% rename from docling/datamodel/pipeline_vlm_model_spec.py rename to docling/datamodel/vlm_model_spec.py diff --git a/docling/utils/model_downloader.py b/docling/utils/model_downloader.py index 585f22a6..8a8e1eb4 100644 --- a/docling/utils/model_downloader.py +++ b/docling/utils/model_downloader.py @@ -6,11 +6,11 @@ from docling.datamodel.pipeline_options import ( granite_picture_description, smolvlm_picture_description, ) -from docling.datamodel.pipeline_vlm_model_spec import ( +from docling.datamodel.settings import settings +from docling.datamodel.vlm_model_spec import ( SMOLDOCLING_MLX, SMOLDOCLING_TRANSFORMERS, ) -from docling.datamodel.settings import settings from docling.models.code_formula_model import CodeFormulaModel from docling.models.document_picture_classifier import DocumentPictureClassifier from docling.models.easyocr_model import EasyOcrModel diff --git a/docs/examples/compare_vlm_models.py b/docs/examples/compare_vlm_models.py index 457c7d58..0ab237ab 100644 --- a/docs/examples/compare_vlm_models.py +++ b/docs/examples/compare_vlm_models.py @@ -16,7 +16,7 @@ from docling.datamodel.base_models import InputFormat from docling.datamodel.pipeline_options import ( VlmPipelineOptions, ) -from docling.datamodel.pipeline_vlm_model_spec import ( +from docling.datamodel.vlm_model_spec import ( GEMMA3_12B_MLX, GRANITE_VISION_OLLAMA, GRANITE_VISION_TRANSFORMERS, diff --git a/docs/examples/minimal_vlm_pipeline.py b/docs/examples/minimal_vlm_pipeline.py index d99a94f8..b0c27eb4 100644 --- a/docs/examples/minimal_vlm_pipeline.py +++ b/docs/examples/minimal_vlm_pipeline.py @@ -2,7 +2,7 @@ from docling.datamodel.base_models import InputFormat from docling.datamodel.pipeline_options import ( VlmPipelineOptions, ) -from docling.datamodel.pipeline_vlm_model_spec import SMOLDOCLING_MLX +from docling.datamodel.vlm_model_spec import SMOLDOCLING_MLX from docling.document_converter import DocumentConverter, PdfFormatOption from docling.pipeline.vlm_pipeline import VlmPipeline