Skip to content

Commit 6b95933

Browse files
committed
Fixed pylint error
Signed-off-by: Cheng, Penghui <penghui.cheng@intel.com>
1 parent 00cd903 commit 6b95933

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

examples/neural_compressor/language-modeling/run_clm.py

+1
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@
5959
from optimum.intel.neural_compressor import INCModelForCausalLM, INCQuantizer, INCTrainer
6060
from optimum.intel.utils.import_utils import is_intel_extension_for_transformers_available
6161

62+
6263
if is_intel_extension_for_transformers_available():
6364
from intel_extension_for_transformers.transformers.utils.config import WeightOnlyQuantConfig
6465

optimum/intel/neural_compressor/quantization.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,7 @@
8080
if is_intel_extension_for_transformers_available():
8181
from intel_extension_for_transformers.llm.quantization.utils import convert_to_quantized_model
8282
from intel_extension_for_transformers.transformers.utils.config import WeightOnlyQuantConfig
83+
8384
Config: TypeAlias = Union[PostTrainingQuantConfig, WeightOnlyQuantConfig]
8485
else:
8586
Config: TypeAlias = PostTrainingQuantConfig
@@ -212,7 +213,9 @@ def quantize(
212213
elif isinstance(quantization_config, WeightOnlyQuantConfig):
213214
algo = quantization_config.algorithm
214215
else:
215-
raise TypeError(f"For weight-only quantization, `quantization_config` should be an instance of `WeightOnlyQuantConfig`, but got: {type(quantization_config)} instead.")
216+
raise TypeError(
217+
f"For weight-only quantization, `quantization_config` should be an instance of `WeightOnlyQuantConfig`, but got: {type(quantization_config)} instead."
218+
)
216219

217220
if calibration_dataset is None and ("GPTQ" in algo or "AWQ" in algo):
218221
raise ValueError(

0 commit comments

Comments
 (0)