From 84683de58d7f4d59a4a692e551c98e4ceff6277b Mon Sep 17 00:00:00 2001 From: Jintao Huang Date: Fri, 5 Jan 2024 17:41:14 +0800 Subject: [PATCH] fix bug --- swift/tuners/utils.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/swift/tuners/utils.py b/swift/tuners/utils.py index 73477936eb..bb4b4d7ba4 100644 --- a/swift/tuners/utils.py +++ b/swift/tuners/utils.py @@ -329,8 +329,7 @@ def freeze_model(): class ModulesToSaveWrapper(ActivationMixin, _ModulesToSaveWrapper): def __init__(self, *args, module_key, **kwargs): - self.module_key = module_key - super(ModulesToSaveWrapper, self).__init__() + super(ModulesToSaveWrapper, self).__init__(module_key) super(ActivationMixin, self).__init__(*args, **kwargs) @property