Skip to content

Commit 9aa6494

Browse files
authored
Update optimum/exporters/openvino/model_configs.py
1 parent 27e8264 commit 9aa6494

1 file changed

Lines changed: 1 addition & 4 deletions

File tree

optimum/exporters/openvino/model_configs.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -587,10 +587,7 @@ class GptOssOpenVINOConfig(LlamaOpenVINOConfig):
587587
library_name="transformers",
588588
)
589589
class BitnetOpenVINOConfig(LlamaOnnxConfig):
590-
def patch_model_for_export(
591-
self, model: Union["PreTrainedModel", "TFPreTrainedModel"], model_kwargs: Optional[Dict[str, Any]] = None
592-
) -> "ModelPatcher":
593-
return OVDecoderModelPatcher(self, model, model_kwargs=model_kwargs)
590+
_MODEL_PATCHER = OVDecoderModelPatcher
594591

595592

596593
@register_in_tasks_manager(

0 commit comments

Comments
 (0)