Update examples/neural_compressor/language-modeling/run_clm.py

Co-authored-by: Ella Charlaix <[email protected]>
PenghuiCheng · Mar 23, 2024 · f51266a · f51266a
1 parent 88760bc
commit f51266a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/neural_compressor/language-modeling/run_clm.py b/examples/neural_compressor/language-modeling/run_clm.py
@@ -228,7 +228,8 @@ class OptimizationArguments:
         default="RTN",
         metadata={"help": "Quantization methodology for weight only quantization. Choose from 'RTN' and 'GPTQ'."},
     )
-    gptq_percdamp: float = field(
+    damp_percent: float = field(
+
         default=0.01,
         metadata={"help": "Percentage of Hessian's diagonal values average, which will be added to Hessian's diagonal to increase numerical stability, used for GPTQ quantization"},
     )