From 07045386c64011f361e5a1e1f3aabd4ec86a6b4b Mon Sep 17 00:00:00 2001 From: Michele Dolfi Date: Mon, 2 Jun 2025 14:29:03 +0200 Subject: [PATCH] remove torch type Signed-off-by: Michele Dolfi --- docling/models/vlm_models_inline/hf_transformers_model.py | 1 - 1 file changed, 1 deletion(-) diff --git a/docling/models/vlm_models_inline/hf_transformers_model.py b/docling/models/vlm_models_inline/hf_transformers_model.py index bc02fbc6..de7f289d 100644 --- a/docling/models/vlm_models_inline/hf_transformers_model.py +++ b/docling/models/vlm_models_inline/hf_transformers_model.py @@ -99,7 +99,6 @@ class HuggingFaceTransformersVlmModel(BasePageModel, HuggingFaceModelDownloadMix self.vlm_model = model_cls.from_pretrained( artifacts_path, device_map=self.device, - torch_dtype="auto", _attn_implementation=( "flash_attention_2" if self.device.startswith("cuda")