From 7d4cb798224ecd47cab2e09c63f6c6e2e43d04db Mon Sep 17 00:00:00 2001
From: hoshi-hiyouga <hiyouga@buaa.edu.cn>
Date: Thu, 26 Sep 2024 11:22:48 +0800
Subject: [PATCH] add modelscope models

Former-commit-id: 4de3081eea9cede78a1f2db65cf22a5731c54447
---
 src/llamafactory/extras/constants.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/llamafactory/extras/constants.py b/src/llamafactory/extras/constants.py
index 2929e7e4..4ff0ce18 100644
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -802,15 +802,19 @@ register_model_group(
     models={
         "LLaMA3.2-1B": {
             DownloadSource.DEFAULT: "meta-llama/Llama-3.2-1B",
+            DownloadSource.MODELSCOPE: "LLM-Research/Llama-3.2-1B",
         },
         "LLaMA3.2-3B": {
             DownloadSource.DEFAULT: "meta-llama/Llama-3.2-3B",
+            DownloadSource.MODELSCOPE: "LLM-Research/Llama-3.2-3B",
         },
         "LLaMA3.2-1B-Instruct": {
             DownloadSource.DEFAULT: "meta-llama/Llama-3.2-1B-Instruct",
+            DownloadSource.MODELSCOPE: "LLM-Research/Llama-3.2-1B-Instruct",
         },
         "LLaMA3.2-3B-Instruct": {
             DownloadSource.DEFAULT: "meta-llama/Llama-3.2-3B-Instruct",
+            DownloadSource.MODELSCOPE: "LLM-Research/Llama-3.2-3B-Instruct",
         },
     },
     template="llama3",