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

Rebase from main

Browse Source 
Signed-off-by: Christoph Auer <cau@zurich.ibm.com>

Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Christoph Auer 2024-12-11 12:30:45 +01:00
commit 038791a25f
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
docling/models/ds_glm_model.py
View File

@ -3,8 +3,7 @@ import random
from pathlib import Path from pathlib import Path
from typing import List, Union from typing import List, Union
from deepsearch_glm.nlp_utils import init_nlp_model from deepsearch_glm.andromeda_nlp import nlp_model
from deepsearch_glm.utils.load_pretrained_models import load_pretrained_nlp_models
from docling_core.types.doc import BoundingBox, CoordOrigin, DoclingDocument from docling_core.types.doc import BoundingBox, CoordOrigin, DoclingDocument
from docling_core.types.legacy_doc.base import BoundingBox as DsBoundingBox from docling_core.types.legacy_doc.base import BoundingBox as DsBoundingBox
from docling_core.types.legacy_doc.base import ( from docling_core.types.legacy_doc.base import (
@ -49,11 +48,7 @@ class GlmModel:
def __init__(self, options: GlmOptions): def __init__(self, options: GlmOptions):
self.options = options self.options = options
if self.options.model_names != "": self.model = nlp_model(loglevel="error", text_ordering=True)
load_pretrained_nlp_models()
self.model = init_nlp_model(
model_names=self.options.model_names, loglevel="ERROR"
)
def _to_legacy_document(self, conv_res) -> DsDocument: def _to_legacy_document(self, conv_res) -> DsDocument:
title = "" title = ""