From 3ac31cf0636bfdcb54899a31f8cef03959117d08 Mon Sep 17 00:00:00 2001
From: DingDing <32740627+ShengdingHu@users.noreply.github.com>
Date: Fri, 28 Oct 2022 15:11:28 +0800
Subject: [PATCH 1/3] Update requirements.txt

---
 requirements.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 12251a9..a4d7b04 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
 torch>=1.8.0
 transformers>=4.10.0
-datasets==1.17.0
+datasets>=1.17.0
 sentencepiece>=0.1.96
 tqdm>=4.62.2
 decorator

From f7d191276716ac16557d845676595299ac06f17b Mon Sep 17 00:00:00 2001
From: DingDing <32740627+ShengdingHu@users.noreply.github.com>
Date: Fri, 28 Oct 2022 15:11:57 +0800
Subject: [PATCH 2/3] Update setup.py

---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 711f5a4..2324fcd 100644
--- a/setup.py
+++ b/setup.py
@@ -6,7 +6,7 @@ import os
 
 requires = """torch>=1.8.0
 transformers>=4.10.0
-datasets==1.17.0
+datasets>=1.17.0
 sentencepiece>=0.1.96
 tqdm>=4.62.2
 decorator
@@ -58,4 +58,4 @@ with open('README.md', 'r') as f:
             "License :: OSI Approved :: Apache Software License",
             "Operating System :: OS Independent",
         ]
-    )
\ No newline at end of file
+    )

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 3/3] 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: