Skip to content

Commit 1fb01db

Browse files
update patcher
1 parent 5a46deb commit 1fb01db

File tree

2 files changed

+414
-14
lines changed

2 files changed

+414
-14
lines changed

optimum/exporters/openvino/model_configs.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@
141141
SanaTextEncoderModelPatcher,
142142
XverseModelPatcher,
143143
Zamba2ModelPatcher,
144-
ZImageTransfromerModelPatcher,
144+
ZImageTransformerModelPatcher,
145145
)
146146

147147

@@ -4558,4 +4558,4 @@ def patch_model_for_export(
45584558
self, model: PreTrainedModel, model_kwargs: Optional[Dict[str, Any]] = None
45594559
) -> ModelPatcher:
45604560
# OpenVINO can not handle complex data in this model
4561-
return ZImageTransfromerModelPatcher(self, model, model_kwargs=model_kwargs)
4561+
return ZImageTransformerModelPatcher(self, model, model_kwargs=model_kwargs)

0 commit comments

Comments
 (0)