mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-27 04:24:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix artifacts_path via pipeline_options

Browse Source 
Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
This commit is contained in:
Michele Dolfi 2024-10-14 08:57:15 +02:00
parent 245b6c4c01
commit ab8f71511b
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
docling/pipeline/standard_pdf_pipeline.py
View File

@ -40,10 +40,11 @@ class StandardPdfPipeline(PaginatedPipeline):
super().__init__(pipeline_options) super().__init__(pipeline_options)
self.pipeline_options: PdfPipelineOptions self.pipeline_options: PdfPipelineOptions
if not pipeline_options.artifacts_path: if pipeline_options.artifacts_path is None:
artifacts_path = self.download_models_hf() self.artifacts_path = self.download_models_hf()
else:
self.artifacts_path = Path(pipeline_options.artifacts_path)
self.artifacts_path = Path(artifacts_path)
self.glm_model = GlmModel( self.glm_model = GlmModel(
options=GlmOptions( options=GlmOptions(
create_legacy_output=pipeline_options.create_legacy_output create_legacy_output=pipeline_options.create_legacy_output