Update code

huggingface · Oct 19, 2023 · 8112179 · 8112179
1 parent 2f9916b
commit 8112179
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/neural_compressor/test_optimization.py b/tests/neural_compressor/test_optimization.py
@@ -56,7 +56,7 @@
     INCSeq2SeqTrainer,
     INCStableDiffusionPipeline,
 )
-from optimum.intel.neural_compressor import WeightOnlyQuantConfig
+from intel_extension_for_transformers.transformers.utils.quantization_config import WeightOnlyQuantConfig
 from optimum.intel.utils.constant import DIFFUSION_WEIGHTS_NAME
 from optimum.onnxruntime import ORTModelForCausalLM, ORTModelForSequenceClassification
 from optimum.pipelines import ORT_SUPPORTED_TASKS