fix phi modules

2024-01-13 23:12:47 +08:00
parent 9aa1a2fc17
commit d1a73fe26c
3 changed files with 7 additions and 6 deletions
--- a/src/llmtuner/extras/constants.py
+++ b/src/llmtuner/extras/constants.py
@@ -443,8 +443,7 @@ register_model_group(
            DownloadSource.DEFAULT: "microsoft/phi-2",
            DownloadSource.MODELSCOPE: "AI-ModelScope/phi-2"
        }
-    },
-    module="Wqkv"
+    }
 )