Skip to content

Commit eab2dbd

Browse files
committed
Fix patcher name
1 parent 46ed113 commit eab2dbd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optimum/exporters/openvino/model_configs.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -610,7 +610,7 @@ class BitnetOpenVINOConfig(LlamaOnnxConfig):
610610
def patch_model_for_export(
611611
self, model: Union["PreTrainedModel", "TFPreTrainedModel"], model_kwargs: Optional[Dict[str, Any]] = None
612612
) -> "ModelPatcher":
613-
return LlamaModelPatcher(self, model, model_kwargs=model_kwargs)
613+
return OVDecoderModelPatcher(self, model, model_kwargs=model_kwargs)
614614

615615

616616
@register_in_tasks_manager(

0 commit comments

Comments
 (0)