From 55fa4eb4e39441289c000c9a2262e8d2e2134759 Mon Sep 17 00:00:00 2001 From: Christoph Auer Date: Wed, 12 Feb 2025 17:09:56 +0100 Subject: [PATCH] Fix repo id Signed-off-by: Christoph Auer --- docling/models/smol_docling_model.py | 2 ++ docs/examples/minimal_smol_docling.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/docling/models/smol_docling_model.py b/docling/models/smol_docling_model.py index 7bbf228a..5e2c7db0 100644 --- a/docling/models/smol_docling_model.py +++ b/docling/models/smol_docling_model.py @@ -28,6 +28,8 @@ _log = logging.getLogger(__name__) class SmolDoclingModel(BasePageModel): + _repo_id: str = "ds4sd/SmolDocling-256M-preview" + def __init__( self, accelerator_options: AcceleratorOptions, diff --git a/docs/examples/minimal_smol_docling.py b/docs/examples/minimal_smol_docling.py index 83eede5f..5d64dee4 100644 --- a/docs/examples/minimal_smol_docling.py +++ b/docs/examples/minimal_smol_docling.py @@ -15,7 +15,7 @@ sources = [ # "tests/data/2305.03393v1-pg9.pdf", ] -pipeline_options = VlmPipelineOptions() # artifacts_path="~/local_model_artifacts/") +pipeline_options = VlmPipelineOptions() # artifacts_path="~/local_model_artifacts/" pipeline_options.generate_page_images = True # If force_backend_text = True, text from backend will be used instead of generated text pipeline_options.force_backend_text = False