From 46700e9f2963b80ee9312e4cb4aa34f0e015f92d Mon Sep 17 00:00:00 2001 From: jane-temcious Date: Thu, 1 May 2025 09:37:27 +0530 Subject: [PATCH] Update picture_description_vlm_model.py Signed-off-by: jane-temcious --- docling/models/picture_description_vlm_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docling/models/picture_description_vlm_model.py b/docling/models/picture_description_vlm_model.py index 374f575d..a88f5532 100644 --- a/docling/models/picture_description_vlm_model.py +++ b/docling/models/picture_description_vlm_model.py @@ -57,7 +57,7 @@ class PictureDescriptionVlmModel(PictureDescriptionBaseModel): artifacts_path, torch_dtype=torch.bfloat16, _attn_implementation=( - "flash_attention_2" if self.device.startswith("cuda") else "eager" + "eager" ), ).to(self.device)