Merge pull request #3683 from YUUUCC/patch-1

Update constants.py to fix the download issue of the Phi-3 model
hiyouga · May 11, 2024 · a384f00 · a384f00
2 parents 101409d + b5c5c31
commit a384f00
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/llmtuner/extras/constants.py b/src/llmtuner/extras/constants.py
@@ -715,11 +715,11 @@ def register_model_group(
     models={
         "Phi3-3.8B-4k-Chat": {
             DownloadSource.DEFAULT: "microsoft/Phi-3-mini-4k-instruct",
-            DownloadSource.DEFAULT: "LLM-Research/Phi-3-mini-4k-instruct",
+            DownloadSource.MODELSCOPE: "LLM-Research/Phi-3-mini-4k-instruct",
         },
         "Phi3-3.8B-128k-Chat": {
             DownloadSource.DEFAULT: "microsoft/Phi-3-mini-128k-instruct",
-            DownloadSource.DEFAULT: "LLM-Research/Phi-3-mini-128k-instruct",
+            DownloadSource.MODELSCOPE: "LLM-Research/Phi-3-mini-128k-instruct",
         },
     },
     module="qkv_proj",