We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 69cf95b commit dc8e3aeCopy full SHA for dc8e3ae
src/llmcompressor/modifiers/awq/base.py
@@ -327,9 +327,7 @@ def _set_resolved_mappings(self, model: Module) -> None:
327
target_patterns = (mapping.smooth_layer, *mapping.balance_layers)
328
329
for modules_set in (
330
- pbar := tqdm(
331
- match_modules_set(model, target_patterns, self.ignore)
332
- )
+ pbar := tqdm(match_modules_set(model, target_patterns, self.ignore))
333
):
334
pbar.set_description(
335
f"Resolving mapping {mapping_idx+1}/{len(self.mappings)}"
0 commit comments