mirror of
https://github.com/DS4SD/docling.git
synced 2025-12-08 12:48:28 +00:00
Add extra_processor_kwargs
Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
@@ -287,6 +287,7 @@ class HuggingFaceTransformersVlmModel(BaseVlmPageModel, HuggingFaceModelDownload
|
||||
return_tensors="pt",
|
||||
padding=True, # pad across batch for both text and vision
|
||||
**self.vlm_options.extra_processor_kwargs,
|
||||
|
||||
)
|
||||
inputs = {k: v.to(self.device) for k, v in inputs.items()}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user