From 42a798d26af4d33c3dc51501768130a0c01bd77d Mon Sep 17 00:00:00 2001
From: DingDing <32740627+ShengdingHu@users.noreply.github.com>
Date: Fri, 28 Oct 2022 15:17:18 +0800
Subject: [PATCH] Update basemodel.py

fix #38
---
 opendelta/basemodel.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/opendelta/basemodel.py b/opendelta/basemodel.py
index cd1006b..b9c3277 100644
--- 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: