Skip to content

Commit 44cdfe9

Browse files
authored
fix tuner bug (#285)
1 parent cf30fa8 commit 44cdfe9

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

swift/tuners/utils.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -329,8 +329,7 @@ def freeze_model():
329329
class ModulesToSaveWrapper(ActivationMixin, _ModulesToSaveWrapper):
330330

331331
def __init__(self, *args, module_key, **kwargs):
332-
self.module_key = module_key
333-
super(ModulesToSaveWrapper, self).__init__()
332+
super(ModulesToSaveWrapper, self).__init__(module_key)
334333
super(ActivationMixin, self).__init__(*args, **kwargs)
335334

336335
@property

0 commit comments

Comments
 (0)