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