Skip to content

Commit 72d51b0

Browse files
style fixes
Signed-off-by: Brian Dellabetta <[email protected]>
1 parent be8bd48 commit 72d51b0

File tree

1 file changed

+1
-2
lines changed
  • src/llmcompressor/modifiers/awq

1 file changed

+1
-2
lines changed

src/llmcompressor/modifiers/awq/base.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -310,8 +310,7 @@ def _set_resolved_mappings(self, model: Module) -> None:
310310
if not balance_layer:
311311
continue
312312

313-
# exclude v_proj/o_proj mappings whose shapes are incompatible
314-
# exclude fused qkv_proj/o_proj mappings when shapes are not all equal
313+
# exclude v_proj->o_proj mappings whose shapes are incompatible
315314
# https://github.com/mit-han-lab/llm-awq/pull/67#issuecomment-1681632777
316315
if (
317316
isinstance(smooth_layer, torch.nn.Linear)

0 commit comments

Comments
 (0)