Fix repo id

Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Christoph Auer 2025-02-12 17:09:56 +01:00 committed by Maksym Lysak
parent 6f9f4f4aee
commit 55fa4eb4e3
2 changed files with 3 additions and 1 deletions

View File

@ -28,6 +28,8 @@ _log = logging.getLogger(__name__)
class SmolDoclingModel(BasePageModel): class SmolDoclingModel(BasePageModel):
_repo_id: str = "ds4sd/SmolDocling-256M-preview"
def __init__( def __init__(
self, self,
accelerator_options: AcceleratorOptions, accelerator_options: AcceleratorOptions,

View File

@ -15,7 +15,7 @@ sources = [
# "tests/data/2305.03393v1-pg9.pdf", # "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 pipeline_options.generate_page_images = True
# If force_backend_text = True, text from backend will be used instead of generated text # If force_backend_text = True, text from backend will be used instead of generated text
pipeline_options.force_backend_text = False pipeline_options.force_backend_text = False