Updated to docling-ibm-models v2.0.0

Signed-off-by: Maxim Lysak <[email protected]>
DS4SD · Oct 3, 2024 · 65ed754 · 65ed754
1 parent 5d5e3ed
commit 65ed754
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/docling/document_converter.py b/docling/document_converter.py
@@ -67,7 +67,10 @@ def download_models_hf(
         from huggingface_hub import snapshot_download
 
         download_path = snapshot_download(
-            repo_id="ds4sd/docling-models", force_download=force, local_dir=local_dir
+            repo_id="ds4sd/docling-models",
+            force_download=force,
+            local_dir=local_dir,
+            revision="v2.0.0",
         )
 
         return Path(download_path)

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -38,7 +38,7 @@ torchvision = [
 python = "^3.10"
 pydantic = "^2.0.0"
 docling-core = "^1.6.2"
-docling-ibm-models = "^1.4.0"
+docling-ibm-models = "^2.0.0"
 deepsearch-glm = "^0.22.0"
 filetype = "^1.2.0"
 pypdfium2 = "^4.30.0"