mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-26 20:14:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove torch type

Browse Source 
Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
This commit is contained in:
Michele Dolfi 2025-06-02 14:29:03 +02:00
parent 738385004a
commit 07045386c6
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -99,7 +99,6 @@ class HuggingFaceTransformersVlmModel(BasePageModel, HuggingFaceModelDownloadMix
self.vlm_model = model_cls.from_pretrained( self.vlm_model = model_cls.from_pretrained(
artifacts_path, artifacts_path,
device_map=self.device, device_map=self.device,
torch_dtype="auto",
_attn_implementation=( _attn_implementation=(
"flash_attention_2" "flash_attention_2"
if self.device.startswith("cuda") if self.device.startswith("cuda")