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

Move imports

Browse Source 
Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Christoph Auer 2025-02-25 14:03:28 +01:00
parent 341806e54b
commit 84c77c9fcb
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docling/models/hf_vlm_model.py
View File

@ -3,8 +3,6 @@ import time
from pathlib import Path from pathlib import Path
from typing import Iterable, List, Optional from typing import Iterable, List, Optional
from transformers import AutoModelForVision2Seq
from docling.datamodel.base_models import Page, VlmPrediction from docling.datamodel.base_models import Page, VlmPrediction
from docling.datamodel.document import ConversionResult from docling.datamodel.document import ConversionResult
from docling.datamodel.pipeline_options import ( from docling.datamodel.pipeline_options import (
@ -36,9 +34,9 @@ class HuggingFaceVlmModel(BasePageModel):
if self.enabled: if self.enabled:
import torch import torch
from transformers import ( # type: ignore from transformers import ( # type: ignore
AutoModelForVision2Seq,
AutoProcessor, AutoProcessor,
BitsAndBytesConfig, BitsAndBytesConfig,
Idefics3ForConditionalGeneration,
) )
device = decide_device(accelerator_options.device) device = decide_device(accelerator_options.device)