File tree Expand file tree Collapse file tree 3 files changed +3
-3
lines changed Expand file tree Collapse file tree 3 files changed +3
-3
lines changed Original file line number Diff line number Diff 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 )
Original file line number Diff line number Diff 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
Original file line number Diff line number Diff line change 2828
2929INSTALL_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" ,
You can’t perform that action at this time.
0 commit comments