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

fix: skip glm model downloads (#322)

Browse Source 
Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
This commit is contained in:
Michele Dolfi 2024-11-13 08:45:28 +01:00 committed by GitHub
parent 2c0c439a44
commit c9341bf22e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
docling/models/ds_glm_model.py
View File

@ -43,6 +43,7 @@ class GlmModel:
def __init__(self, options: GlmOptions): def __init__(self, options: GlmOptions):
self.options = options self.options = options
if self.options.model_names != "":
load_pretrained_nlp_models() load_pretrained_nlp_models()
self.model = init_nlp_model(model_names=self.options.model_names) self.model = init_nlp_model(model_names=self.options.model_names)