Skip to content

Commit 0b4c6a5

Browse files
gavingavin99gavinlee
andauthored
fix bug:makes transform ignore filtering more flexible (#274)
Co-authored-by: gavinlee <gavinlee@example.com>
1 parent bf884ae commit 0b4c6a5

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

angelslim/models/base_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -343,7 +343,7 @@ def get_rotation_mapping_layers(
343343
for name, module in self.model.named_modules():
344344
for target in targets:
345345
if name.split(".")[-1] == target:
346-
if name not in ignore:
346+
if not any(name.endswith(ig) for ig in ignore):
347347
linear_mapping_layers[name] = module
348348
break
349349
return linear_mapping_layers

0 commit comments

Comments
 (0)