mirror of
https://github.com/DS4SD/docling.git
synced 2025-07-26 20:14:47 +00:00
fix mypy
Signed-off-by: Georg Heiler <georg.kf.heiler@gmail.com>
This commit is contained in:
parent
448c932fd2
commit
5ce2892940
@ -94,7 +94,7 @@ class HuggingFaceTransformersVlmModel(BasePageModel, HuggingFaceModelDownloadMix
|
||||
model_cls = AutoModelForVision2Seq
|
||||
elif (
|
||||
self.vlm_options.transformers_model_type
|
||||
== TransformersModelType.AUTOMODEL_FORIMAGETEXTTOTEXT
|
||||
== TransformersModelType.AUTOMODEL_IMAGETEXTTOTEXT
|
||||
):
|
||||
model_cls = AutoModelForImageTextToText
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user