NeoAnd/Docling
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: new torch-based docling models (#120)

Browse Source 
---------

Signed-off-by: Maxim Lysak <mly@zurich.ibm.com>
Co-authored-by: Maxim Lysak <mly@zurich.ibm.com>
This commit is contained in:
Maxim Lysak
2024-10-03 18:42:33 +02:00
committed by GitHub
parent 9ebbbc1245
commit 2422f706a1
30 changed files with 1159 additions and 1185 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docling/pipeline/standard_model_pipeline.py
View File

@@ -8,7 +8,7 @@ from docling.pipeline.base_model_pipeline import BaseModelPipeline
class StandardModelPipeline(BaseModelPipeline):
_layout_model_path = "model_artifacts/layout/beehive_v0.0.5"
_layout_model_path = "model_artifacts/layout/beehive_v0.0.5_pt"
_table_model_path = "model_artifacts/tableformer"
def __init__(self, artifacts_path: Path, pipeline_options: PipelineOptions):