Skip to content

Commit 0a09651

Browse files
authored
update optimum version for having latest fixes (#1085)
* update optimum version for having latest fixes * update imports
1 parent 420fa87 commit 0a09651

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

optimum/exporters/openvino/convert.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -365,7 +365,7 @@ def export_pytorch(
365365
import torch
366366
from torch.utils._pytree import tree_map
367367

368-
from optimum.exporters.onnx.convert import check_dummy_inputs_are_allowed
368+
from optimum.exporters.utils import check_dummy_inputs_are_allowed
369369

370370
logger.info(f"Using framework PyTorch: {torch.__version__}")
371371
output = Path(output)

optimum/intel/openvino/quantization.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -523,7 +523,7 @@ def _quantize_torchmodel(
523523

524524
quantization_config = ov_config.quantization_config
525525
if isinstance(quantization_config, OVWeightQuantizationConfig):
526-
from optimum.exporters.onnx.convert import check_dummy_inputs_are_allowed
526+
from optimum.exporters.utils import check_dummy_inputs_are_allowed
527527

528528
if stateful:
529529
# patch model before weight compression

setup.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828

2929
INSTALL_REQUIRE = [
3030
"torch>=1.11",
31-
"optimum~=1.23",
31+
"optimum@git+https://github.com/huggingface/optimum.git",
3232
"transformers>=4.36,<4.47",
3333
"datasets>=1.4.0",
3434
"sentencepiece",

0 commit comments

Comments
 (0)