Skip to content

Commit 11f31f5

Browse files
authored
Update smoothquant.py (#191)
1 parent dba7c55 commit 11f31f5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

llmc/compression/quantization/smoothquant.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def subset_transform(
7171
inspect_module,
7272
subset_kwargs,
7373
):
74-
if not self.filter_subset(prev_op):
74+
if not self.filter_subset(layers_dict, prev_op):
7575
logger.info('Do not transform this subset.')
7676
return
7777
layers = list(layers_dict.values())

0 commit comments

Comments
 (0)