Skip to content

Commit 04ada71

Browse files
exclude observer modules
Signed-off-by: Brian Dellabetta <[email protected]>
1 parent 9dab4dd commit 04ada71

File tree

1 file changed

+3
-0
lines changed
  • src/llmcompressor/modifiers/awq

1 file changed

+3
-0
lines changed

src/llmcompressor/modifiers/awq/base.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -332,6 +332,9 @@ def _set_resolved_mappings(self, model: Module) -> None:
332332
balance_regex,
333333
smooth_parent,
334334
).items():
335+
if balance_suffix.endswith("observer"):
336+
continue
337+
335338
balance_name = f"{smooth_parent_name}.{balance_suffix}"
336339

337340
# exclude v_proj->o_proj mappings whose shapes are incompatible

0 commit comments

Comments
 (0)