fix

Signed-off-by: yiliu30 <[email protected]>

Author: yiliu30

examples/pytorch/nlp/huggingface_models/language-modeling/quantization/auto_round/quantize.py

@@ -32,7 +32,7 @@
 }
 
 
-def quant_model(args):
+def quant_model_ar(args):
     config = topologies_config[args.t]
 
     logger.info(f"Using fp_layers: {config['fp_layers']}")
@@ -65,7 +65,7 @@ def get_model_and_tokenizer(model_name):
     return fp32_model, tokenizer
 
 
-def quant_model_(args):
+def quant_model(args):
     from neural_compressor.torch.quantization import (
         AutoRoundConfig,
         convert,