From 84c77c9fcb7608278ba648209eabea8912000866 Mon Sep 17 00:00:00 2001
From: Christoph Auer <cau@zurich.ibm.com>
Date: Tue, 25 Feb 2025 14:03:28 +0100
Subject: [PATCH] Move imports

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

diff --git a/docling/models/hf_vlm_model.py b/docling/models/hf_vlm_model.py
index 973cb374..619658d3 100644
--- a/docling/models/hf_vlm_model.py
+++ b/docling/models/hf_vlm_model.py
@@ -3,8 +3,6 @@ import time
 from pathlib import Path
 from typing import Iterable, List, Optional
 
-from transformers import AutoModelForVision2Seq
-
 from docling.datamodel.base_models import Page, VlmPrediction
 from docling.datamodel.document import ConversionResult
 from docling.datamodel.pipeline_options import (
@@ -36,9 +34,9 @@ class HuggingFaceVlmModel(BasePageModel):
         if self.enabled:
             import torch
             from transformers import (  # type: ignore
+                AutoModelForVision2Seq,
                 AutoProcessor,
                 BitsAndBytesConfig,
-                Idefics3ForConditionalGeneration,
             )
 
             device = decide_device(accelerator_options.device)