update optimum version for having latest fixes (#1085)

* update optimum version for having latest fixes * update imports
huggingface · Dec 20, 2024 · 0a09651 · 0a09651
1 parent 420fa87
commit 0a09651
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/optimum/exporters/openvino/convert.py b/optimum/exporters/openvino/convert.py
@@ -365,7 +365,7 @@ def export_pytorch(
     import torch
     from torch.utils._pytree import tree_map
 
-    from optimum.exporters.onnx.convert import check_dummy_inputs_are_allowed
+    from optimum.exporters.utils import check_dummy_inputs_are_allowed
 
     logger.info(f"Using framework PyTorch: {torch.__version__}")
     output = Path(output)

diff --git a/optimum/intel/openvino/quantization.py b/optimum/intel/openvino/quantization.py
@@ -523,7 +523,7 @@ def _quantize_torchmodel(
 
         quantization_config = ov_config.quantization_config
         if isinstance(quantization_config, OVWeightQuantizationConfig):
-            from optimum.exporters.onnx.convert import check_dummy_inputs_are_allowed
+            from optimum.exporters.utils import check_dummy_inputs_are_allowed
 
             if stateful:
                 # patch model before weight compression

diff --git a/setup.py b/setup.py
@@ -28,7 +28,7 @@
 
 INSTALL_REQUIRE = [
     "torch>=1.11",
-    "optimum~=1.23",
+    "optimum@git+https://github.com/huggingface/optimum.git",
     "transformers>=4.36,<4.47",
     "datasets>=1.4.0",
     "sentencepiece",