From 448c932fd2f2d991b3daed9d7ba48429fda758e7 Mon Sep 17 00:00:00 2001 From: Georg Heiler Date: Tue, 17 Jun 2025 08:23:39 +0200 Subject: [PATCH] reformat Signed-off-by: Georg Heiler --- docling/models/vlm_models_inline/hf_transformers_model.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/docling/models/vlm_models_inline/hf_transformers_model.py b/docling/models/vlm_models_inline/hf_transformers_model.py index 73dd439b..18bd7d86 100644 --- a/docling/models/vlm_models_inline/hf_transformers_model.py +++ b/docling/models/vlm_models_inline/hf_transformers_model.py @@ -41,8 +41,8 @@ class HuggingFaceTransformersVlmModel(BasePageModel, HuggingFaceModelDownloadMix from transformers import ( AutoModel, AutoModelForCausalLM, - AutoModelForVision2Seq, AutoModelForImageTextToText, + AutoModelForVision2Seq, AutoProcessor, BitsAndBytesConfig, GenerationConfig, @@ -92,8 +92,10 @@ class HuggingFaceTransformersVlmModel(BasePageModel, HuggingFaceModelDownloadMix == TransformersModelType.AUTOMODEL_VISION2SEQ ): model_cls = AutoModelForVision2Seq - elif (self.vlm_options.transformers_model_type - == TransformersModelType.AUTOMODEL_FORIMAGETEXTTOTEXT): + elif ( + self.vlm_options.transformers_model_type + == TransformersModelType.AUTOMODEL_FORIMAGETEXTTOTEXT + ): model_cls = AutoModelForImageTextToText self.processor = AutoProcessor.from_pretrained(