mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-08-01 15:02:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: fix leftover .to(device)

Browse Source 
Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Christoph Auer 2025-02-26 12:53:43 +01:00
parent 5cd0fdd258
commit b88440a7c9
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/hf_vlm_model.py
View File

@ -85,7 +85,7 @@ class HuggingFaceVlmModel(BasePageModel):
and accelerator_options.cuda_use_flash_attention2 and accelerator_options.cuda_use_flash_attention2
else "eager" else "eager"
), ),
).to(self.device) ) # .to(self.device)
@staticmethod @staticmethod
def download_models( def download_models(