fix setup

Author: echarlaix

optimum/intel/openvino/trainer.py

@@ -905,7 +905,7 @@ def _save(self, output_dir: Optional[str] = None, state_dict=None):
             self.compression_controller.prepare_for_export()
             model_type = self.model.config.model_type.replace("_", "-")
             exporter_config_class = TasksManager.get_exporter_config_constructor(
-                exporter="openvino",
+                exporter="onnx",
                 model=self.model,
                 task=self.task,
                 model_type=model_type,

setup.py

@@ -28,8 +28,8 @@
 
 INSTALL_REQUIRE = [
     "torch>=1.11",
-    "transformers>=4.36.0,<4.42.0",
-    "optimum @ git+https://github.com/huggingface/optimum.git@bump-transformers",
+    "transformers>=4.36.0,<4.41.0",
+    "optimum",
     "datasets>=1.4.0",
     "sentencepiece",
     "scipy",