mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-27 04:24:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update picture_description_vlm_model.py

Browse Source 
Signed-off-by: jane-temcious <jane.temcious@iterate.ai>
This commit is contained in:
jane-temcious 2025-05-01 09:37:27 +05:30 committed by GitHub
parent 8012a3e4d6
commit 46700e9f29
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docling/models/picture_description_vlm_model.py
View File

@ -57,7 +57,7 @@ class PictureDescriptionVlmModel(PictureDescriptionBaseModel):
artifacts_path, artifacts_path,
torch_dtype=torch.bfloat16, torch_dtype=torch.bfloat16,
_attn_implementation=( _attn_implementation=(
"flash_attention_2" if self.device.startswith("cuda") else "eager" "eager"
), ),
).to(self.device) ).to(self.device)