From b88440a7c93d4b5721617be836f578bc4c1ba518 Mon Sep 17 00:00:00 2001
From: Christoph Auer <cau@zurich.ibm.com>
Date: Wed, 26 Feb 2025 12:53:43 +0100
Subject: [PATCH] chore: fix leftover .to(device)

Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
---
 docling/models/hf_vlm_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docling/models/hf_vlm_model.py b/docling/models/hf_vlm_model.py
index f0280cc6..2acbe290 100644
--- a/docling/models/hf_vlm_model.py
+++ b/docling/models/hf_vlm_model.py
@@ -85,7 +85,7 @@ class HuggingFaceVlmModel(BasePageModel):
                         and accelerator_options.cuda_use_flash_attention2
                         else "eager"
                     ),
-                ).to(self.device)
+                )  # .to(self.device)
 
     @staticmethod
     def download_models(