Merge branch 'main' of github.com:thunlp/OpenDelta into main

2022-11-04 07:18:22 +00:00 · 2022-11-04 07:18:22 +00:00 · fd8ead687a
parent b0e7e5b619 42a798d26a
commit fd8ead687a
1 changed files with 2 additions and 0 deletions
--- a/opendelta/basemodel.py
+++ b/opendelta/basemodel.py
@ -140,6 +140,8 @@ class DeltaBase(nn.Module, SaveLoadMixin):
            self.structure_mapping = None
        if unfrozen_modules is None:
            self.unfrozen_modules = self.default_unfrozen_modules
+        else:
+            self.unfrozen_modules = unfrozen_modules
        if self.common_structure and self.structure_mapping is None:
            raise RuntimeError("Using common structure but the structure mapping is None")
        if backend not in self._supported_backends: