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

fix usage of artifacts path

Browse Source 
Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
This commit is contained in:
Michele Dolfi 2025-02-25 14:09:14 +01:00
parent c84b973959
commit 96d9fe19cd
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docling/models/picture_description_vlm_model.py
View File

@ -41,9 +41,9 @@ class PictureDescriptionVlmModel(PictureDescriptionBaseModel):
) )
# Initialize processor and model # Initialize processor and model
self.processor = AutoProcessor.from_pretrained(self.options.repo_id) self.processor = AutoProcessor.from_pretrained(artifacts_path)
self.model = AutoModelForVision2Seq.from_pretrained( self.model = AutoModelForVision2Seq.from_pretrained(
self.options.repo_id, artifacts_path,
torch_dtype=torch.bfloat16, torch_dtype=torch.bfloat16,
_attn_implementation=( _attn_implementation=(
"flash_attention_2" if self.device.startswith("cuda") else "eager" "flash_attention_2" if self.device.startswith("cuda") else "eager"