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

PenghuiCheng · echarlaix · web-flow · commit f51266ad480a · 2024-03-23T21:15:51.000+08:00
Co-authored-by: Ella Charlaix &lt;80481427+echarlaix@users.noreply.github.com&gt;
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"},
     )

Original file line number	Diff line number	Diff line change
`@@ -228,7 +228,8 @@ class OptimizationArguments:`
`228`	`228`	`default="RTN",`
`229`	`229`	`metadata={"help": "Quantization methodology for weight only quantization. Choose from 'RTN' and 'GPTQ'."},`
`230`	`230`	`)`
`231`		`- gptq_percdamp: float = field(`
	`231`	`+ damp_percent: float = field(`
	`232`	`+`
`232`	`233`	`default=0.01,`
`233`	`234`	`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"},`
`234`	`235`	`)`