From dbb35c7f282dde466d184e96b1d68880ff520b6d Mon Sep 17 00:00:00 2001 From: Michele Dolfi Date: Fri, 7 Feb 2025 14:01:39 +0100 Subject: [PATCH] use with_smolvlm in models download Signed-off-by: Michele Dolfi --- docling/cli/models.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/docling/cli/models.py b/docling/cli/models.py index aea498c5..3b62ad6b 100644 --- a/docling/cli/models.py +++ b/docling/cli/models.py @@ -31,6 +31,7 @@ class _AvailableModels(str, Enum): TABLEFORMER = "tableformer" CODE_FORMULA = "code_formula" PICTURE_CLASSIFIER = "picture_classifier" + SMOLVLM = "smolvlm" EASYOCR = "easyocr" @@ -81,6 +82,7 @@ def download( with_tableformer=_AvailableModels.TABLEFORMER in to_download, with_code_formula=_AvailableModels.CODE_FORMULA in to_download, with_picture_classifier=_AvailableModels.PICTURE_CLASSIFIER in to_download, + with_smolvlm=_AvailableModels.SMOLVLM in to_download, with_easyocr=_AvailableModels.EASYOCR in to_download, )