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

Add extra_processor_kwargs

Browse Source 
Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Christoph Auer
2025-08-20 14:19:50 +02:00
parent 6bbb8e6340 b5b7e6dd5c
commit 684adc17df
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docling/models/vlm_models_inline/hf_transformers_model.py
View File

@@ -287,6 +287,7 @@ class HuggingFaceTransformersVlmModel(BaseVlmPageModel, HuggingFaceModelDownload
return_tensors="pt", return_tensors="pt",
padding=True, # pad across batch for both text and vision padding=True, # pad across batch for both text and vision
**self.vlm_options.extra_processor_kwargs, **self.vlm_options.extra_processor_kwargs,
) )
inputs = {k: v.to(self.device) for k, v in inputs.items()} inputs = {k: v.to(self.device) for k, v in inputs.items()}