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

add message for transformers version

Browse Source 
Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
This commit is contained in:
Michele Dolfi 2025-06-02 09:55:15 +02:00
parent 3ba698984d
commit 76718cb1f9
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
docling/models/vlm_models_inline/hf_transformers_causallm_model.py
View File

@ -1,3 +1,4 @@
import importlib.metadata
import logging import logging
import time import time
from collections.abc import Iterable from collections.abc import Iterable
@ -39,6 +40,15 @@ class HuggingFaceVlmModel_AutoModelForCausalLM(BasePageModel):
GenerationConfig, GenerationConfig,
) )
transformers_version = importlib.metadata.version("transformers")
if (
self.vlm_options.repo_id == "microsoft/Phi-4-multimodal-instruct"
and transformers_version >= "4.52.0"
):
raise NotImplementedError(
f"Phi 4 only works with transformers<4.52.0 but you have {transformers_version=}. Please downgrage running pip install -U 'transformers<4.52.0'."
)
self.device = decide_device( self.device = decide_device(
accelerator_options.device, accelerator_options.device,
supported_devices=vlm_options.supported_devices, supported_devices=vlm_options.supported_devices,