add llama3 chinese chat

hiyouga · May 8, 2024 · 48ee46d · 48ee46d
1 parent 10ab83f
commit 48ee46d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/llmtuner/extras/constants.py b/src/llmtuner/extras/constants.py
@@ -577,6 +577,10 @@ def register_model_group(
             DownloadSource.DEFAULT: "meta-llama/Meta-Llama-3-70B-Instruct",
             DownloadSource.MODELSCOPE: "LLM-Research/Meta-Llama-3-70B-Instruct",
         },
+        "LLaMA3-8B-Chinese-Chat": {
+            DownloadSource.DEFAULT: "shenzhi-wang/Llama3-8B-Chinese-Chat",
+            DownloadSource.MODELSCOPE: "LLM-Research/Llama3-8B-Chinese-Chat",
+        },
     },
     template="llama3",
 )