From 76718cb1f9a8e235847734bc7d6ab480e31c5ba4 Mon Sep 17 00:00:00 2001 From: Michele Dolfi Date: Mon, 2 Jun 2025 09:55:15 +0200 Subject: [PATCH] add message for transformers version Signed-off-by: Michele Dolfi --- .../hf_transformers_causallm_model.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/docling/models/vlm_models_inline/hf_transformers_causallm_model.py b/docling/models/vlm_models_inline/hf_transformers_causallm_model.py index 9d331e8b..d3d6a93d 100644 --- a/docling/models/vlm_models_inline/hf_transformers_causallm_model.py +++ b/docling/models/vlm_models_inline/hf_transformers_causallm_model.py @@ -1,3 +1,4 @@ +import importlib.metadata import logging import time from collections.abc import Iterable @@ -39,6 +40,15 @@ class HuggingFaceVlmModel_AutoModelForCausalLM(BasePageModel): 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( accelerator_options.device, supported_devices=vlm_options.supported_devices,