Skip to content

Commit 2bd08f6

Browse files
committed
fix exporter
1 parent 63bf2ab commit 2bd08f6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optimum/intel/openvino/trainer.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -907,7 +907,7 @@ def _save(self, output_dir: Optional[str] = None, state_dict=None):
907907
self.compression_controller.prepare_for_export()
908908
model_type = self.model.config.model_type.replace("_", "-")
909909
exporter_config_class = TasksManager.get_exporter_config_constructor(
910-
exporter="openvino",
910+
exporter="onnx",
911911
model=self.model,
912912
task=self.task,
913913
model_type=model_type,

0 commit comments

Comments
 (0)