fix typo

echarlaix · echarlaix · commit 7b4c8d9d0927 · 2024-05-14T18:01:24.000+02:00
diff --git a/optimum/intel/neural_compressor/modeling_base.py b/optimum/intel/neural_compressor/modeling_base.py
@@ -147,7 +147,7 @@ def _from_pretrained(
             try:
                 quantization_config = PretrainedConfig.from_pretrained(model_save_dir / "quantize_config.json")
                 algorithm = getattr(quantization_config, "quant_method", None)
-                if algorithm in {"rtn", "gptq", "awq", "autoaround"}:
+                if algorithm in {"rtn", "gptq", "awq", "autoround"}:
                     from intel_extension_for_transformers.transformers.modeling.modeling_auto import (
                         _BaseQBitsAutoModelClass,
                     )

Original file line number	Diff line number	Diff line change
`@@ -147,7 +147,7 @@ def _from_pretrained(`
`147`	`147`	`try:`
`148`	`148`	`quantization_config = PretrainedConfig.from_pretrained(model_save_dir / "quantize_config.json")`
`149`	`149`	`algorithm = getattr(quantization_config, "quant_method", None)`
`150`		`- if algorithm in {"rtn", "gptq", "awq", "autoaround"}:`
	`150`	`+ if algorithm in {"rtn", "gptq", "awq", "autoround"}:`
`151`	`151`	`from intel_extension_for_transformers.transformers.modeling.modeling_auto import (`
`152`	`152`	`_BaseQBitsAutoModelClass,`
`153`	`153`	`)`