mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-08-02 07:22:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: Silence the tqdm messages during the downloading of model files

Browse Source 
Signed-off-by: Nikos Livathinos <nli@zurich.ibm.com>
This commit is contained in:
Nikos Livathinos 2024-12-08 18:35:06 +01:00
parent e125b9b24d
commit 64c7382880
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
docling/pipeline/standard_pdf_pipeline.py
View File

@ -98,6 +98,11 @@ class StandardPdfPipeline(PaginatedPipeline):
) -> Path: ) -> Path:
from huggingface_hub import snapshot_download from huggingface_hub import snapshot_download
# Disable tqdm prints used by HF
from tqdm import tqdm
from functools import partialmethod
tqdm.__init__ = partialmethod(tqdm.__init__, disable=True)
download_path = snapshot_download( download_path = snapshot_download(
repo_id="ds4sd/docling-models", repo_id="ds4sd/docling-models",
force_download=force, force_download=force,