Skip to content

Commit 5ffc4bc

Browse files
committed
Fix smoothq_scale_layers key name
Signed-off-by: Andrea Fasoli <[email protected]>
1 parent 6ee7e4c commit 5ffc4bc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

fms_mo/utils/dq_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ def config_quantize_smooth_layers(qcfg: dict):
117117
qcfg["smoothq_act_scale_path"] = "./act_scales/graniteCodeHF_34b_base12.pt"
118118
elif "roberta" in qcfg["model"]:
119119
qcfg["act_scale_path"] = "./act_scales"
120-
qcfg["scale_layers"] = [
120+
qcfg["smoothq_scale_layers"] = [
121121
"attention.self.query",
122122
"attention.self.key",
123123
"attention.self.value",

0 commit comments

Comments
 (0)