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
|
model_cls = AutoModelForVision2Seq
|
||||||
elif (
|
elif (
|
||||||
self.vlm_options.transformers_model_type
|
self.vlm_options.transformers_model_type
|
||||||
== TransformersModelType.AUTOMODEL_FORIMAGETEXTTOTEXT
|
== TransformersModelType.AUTOMODEL_IMAGETEXTTOTEXT
|
||||||
):
|
):
|
||||||
model_cls = AutoModelForImageTextToText
|
model_cls = AutoModelForImageTextToText
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user