diff --git a/.github/workflows/test_inc.yml b/.github/workflows/test_inc.yml index 63ceb75158..3ba52ec008 100644 --- a/.github/workflows/test_inc.yml +++ b/.github/workflows/test_inc.yml @@ -30,7 +30,7 @@ jobs: - name: Install dependencies run: | python -m pip install --upgrade pip - pip install cmake>=3.16 + pip install cmake pip install py-cpuinfo pip install torch==2.1.0 torchaudio==2.1.0 torchvision==0.16 --extra-index-url https://download.pytorch.org/whl/cpu pip install .[neural-compressor,diffusers,tests] diff --git a/optimum/intel/neural_compressor/quantization.py b/optimum/intel/neural_compressor/quantization.py index d5ff782db3..9099efcf74 100644 --- a/optimum/intel/neural_compressor/quantization.py +++ b/optimum/intel/neural_compressor/quantization.py @@ -214,7 +214,7 @@ def quantize( if quantization_config is None: quantization_config = WeightOnlyQuantConfig() - algo = ["RTN"] + algo = "RTN" elif isinstance(quantization_config, WeightOnlyQuantConfig): algo = quantization_config.algorithm else: @@ -223,7 +223,7 @@ def quantize( ) if algo not in ["RTN", "GPTQ"]: - raise ValueError("Weight-only quantization is only support RTN and GPTQ algorithm now!") + raise ValueError(f"Weight-only quantization is only support RTN and GPTQ algorithm now!But got {algo}") if calibration_dataset is None and quantization_config.tokenizer is None and ("GPTQ" in algo): raise ValueError(