Skip to content

Commit f65bb8f

Browse files
committed
Fix line length issue
1 parent 774c906 commit f65bb8f

2 files changed

Lines changed: 3 additions & 4 deletions

File tree

src/llmcompressor/modifiers/awq/mappings.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -181,8 +181,9 @@ class AWQMapping:
181181
),
182182
]
183183

184-
# AFMOE uses dual normalization: pre_mlp_layernorm feeds the MLP (not post_attention_layernorm)
185-
# and attention has its own gate_proj for gating mechanism
184+
# AFMOE uses dual normalization: pre_mlp_layernorm feeds the MLP
185+
# (not post_attention_layernorm) and attention has its own gate_proj
186+
# for gating mechanism
186187
_afmoe_mappings = [
187188
AWQMapping(
188189
"re:.*input_layernorm$",

src/llmcompressor/modifiers/smoothquant/utils.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -66,8 +66,6 @@
6666
),
6767
]
6868

69-
# AFMOE uses dual normalization: pre_mlp_layernorm feeds the MLP (not post_attention_layernorm)
70-
# and attention has its own gate_proj for gating mechanism
7169
AFMOE_SMOOTHQUANT_MAPPINGS: list[LayerMap] = [
7270
LayerMap(
7371
balance_layers=[

0 commit comments

Comments
 (0)