fix hiyouga#2481

xin-li-67 · Feb 15, 2024 · 22acab8 · 22acab8
1 parent 7924ffc
commit 22acab8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/data/dataset_info.json b/data/dataset_info.json
@@ -190,7 +190,7 @@
     "tags": {
       "role_tag": "role",
       "content_tag": "content",
-      "user_tag": "human",
+      "user_tag": "user",
       "assistant_tag": "assistant"
     },
     "formatting": "sharegpt"

diff --git a/src/llmtuner/model/adapter.py b/src/llmtuner/model/adapter.py
@@ -63,8 +63,8 @@ def init_adapter(
 
         freeze_modules = {"all"}
         for name, _ in model.named_modules():
-            if "0." in name:
-                freeze_modules.add(name.split("0.")[-1].split(".")[0])
+            if ".0." in name:
+                freeze_modules.add(name.split(".0.")[-1].split(".")[0])
 
         trainable_layers = []
         for module_name in finetuning_args.name_module_trainable: