Update src/llmcompressor/modifiers/awq/base.py

ved1beta · kylesayrs · web-flow · commit 94d2b943f66b · 2025-07-30T11:47:59.000+05:30
Co-authored-by: Kyle Sayers &lt;kylesayrs@gmail.com&gt;
diff --git a/src/llmcompressor/modifiers/awq/base.py b/src/llmcompressor/modifiers/awq/base.py
@@ -306,7 +306,7 @@ def _set_resolved_mappings(self, model: Module) -> None:
         resolved_mappings: list[ResolvedMapping] = []
         for mapping_idx, mapping in enumerate(self.mappings):
             smooth_layers = match_named_modules(
-                mapping.smooth_layer, model, exclude_internal_modules=True
+                model, [mapping.smooth_layer]
             )
             smooth_names = [
                 smooth_name

Original file line number	Diff line number	Diff line change
`@@ -306,7 +306,7 @@ def _set_resolved_mappings(self, model: Module) -> None:`
`306`	`306`	`resolved_mappings: list[ResolvedMapping] = []`
`307`	`307`	`for mapping_idx, mapping in enumerate(self.mappings):`
`308`	`308`	`smooth_layers = match_named_modules(`
`309`		`- mapping.smooth_layer, model, exclude_internal_modules=True`
	`309`	`+ model, [mapping.smooth_layer]`
`310`	`310`	`)`
`311`	`311`	`smooth_names = [`
`312`	`312`	`smooth_name`