Skip to content

Commit 8e4a81e

Browse files
committed
Make check dummy inputs import conditional in quantizer
1 parent a76be08 commit 8e4a81e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

optimum/intel/openvino/quantization.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,6 @@
4444
from transformers.pytorch_utils import Conv1D
4545
from transformers.utils import is_accelerate_available
4646

47-
from optimum.exporters.onnx.convert import check_dummy_inputs_are_allowed
4847
from optimum.exporters.tasks import TasksManager
4948
from optimum.quantization_base import OptimumQuantizer
5049

@@ -524,6 +523,8 @@ def _quantize_torchmodel(
524523

525524
quantization_config = ov_config.quantization_config
526525
if isinstance(quantization_config, OVWeightQuantizationConfig):
526+
from optimum.exporters.onnx.convert import check_dummy_inputs_are_allowed
527+
527528
if stateful:
528529
# patch model before weight compression
529530
model = patch_model_with_bettertransformer(model)

0 commit comments

Comments
 (0)