add llama3 chinese chat

2024-05-08 17:10:03 +08:00 · 2024-05-08 17:10:03 +08:00 · 48ee46dac1
parent 10ab83f4c4
commit 48ee46dac1
1 changed files with 4 additions and 0 deletions
--- a/src/llmtuner/extras/constants.py
+++ b/src/llmtuner/extras/constants.py
@ -577,6 +577,10 @@ 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",
 )