File tree 3 files changed +3
-3
lines changed
3 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -365,7 +365,7 @@ def export_pytorch(
365
365
import torch
366
366
from torch .utils ._pytree import tree_map
367
367
368
- from optimum .exporters .onnx . convert import check_dummy_inputs_are_allowed
368
+ from optimum .exporters .utils import check_dummy_inputs_are_allowed
369
369
370
370
logger .info (f"Using framework PyTorch: { torch .__version__ } " )
371
371
output = Path (output )
Original file line number Diff line number Diff line change @@ -523,7 +523,7 @@ def _quantize_torchmodel(
523
523
524
524
quantization_config = ov_config .quantization_config
525
525
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
527
527
528
528
if stateful :
529
529
# patch model before weight compression
Original file line number Diff line number Diff line change 28
28
29
29
INSTALL_REQUIRE = [
30
30
"torch>=1.11" ,
31
- "optimum~=1.23 " ,
31
+ "optimum@git+https://github.com/huggingface/optimum.git " ,
32
32
"transformers>=4.36,<4.47" ,
33
33
"datasets>=1.4.0" ,
34
34
"sentencepiece" ,
You can’t perform that action at this time.
0 commit comments