fix phi modules

Former-commit-id: 68d7e925ec51b6ee277513de8f61ac18a8378b98
2024-01-13 23:12:47 +08:00
parent c5f099138d
commit ad99bd0a14
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"
+    }
 )