We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent bf884ae commit 0b4c6a5Copy full SHA for 0b4c6a5
1 file changed
angelslim/models/base_model.py
@@ -343,7 +343,7 @@ def get_rotation_mapping_layers(
343
for name, module in self.model.named_modules():
344
for target in targets:
345
if name.split(".")[-1] == target:
346
- if name not in ignore:
+ if not any(name.endswith(ig) for ig in ignore):
347
linear_mapping_layers[name] = module
348
break
349
return linear_mapping_layers
0 commit comments