p04798526
/
LLaMA-Factory-310P3
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add model

This commit is contained in:
tastelikefeet 2023-12-01 15:06:17 +08:00
parent fb2204c183
commit 8ce4d11e38
1 changed files with 21 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
src/llmtuner/extras/constants.py
View File

@ -107,11 +107,11 @@ register_model_group(
}, },
"BLOOM-3B": { "BLOOM-3B": {
"hf": "bigscience/bloom-3b", "hf": "bigscience/bloom-3b",
"ms": "bigscience/bloom-3b", "ms": "AI-ModelScope/bloom-3b",
}, },
"BLOOM-7B1": { "BLOOM-7B1": {
"hf": "bigscience/bloom-7b1", "hf": "bigscience/bloom-7b1",
"ms": "bigscience/bloom-7b1", "ms": "AI-ModelScope/bloom-7b1",
} }
}, },
module="query_key_value" module="query_key_value"
@ -122,11 +122,11 @@ register_model_group(
models={ models={
"BLOOMZ-560M": { "BLOOMZ-560M": {
"hf": "bigscience/bloomz-560m", "hf": "bigscience/bloomz-560m",
"ms": "bigscience/bloomz-560m", "ms": "AI-ModelScope/bloomz-560m",
}, },
"BLOOMZ-3B": { "BLOOMZ-3B": {
"hf": "bigscience/bloomz-3b", "hf": "bigscience/bloomz-3b",
"ms": "bigscience/bloomz-3b", "ms": "AI-ModelScope/bloomz-3b",
}, },
"BLOOMZ-7B1-mt": { "BLOOMZ-7B1-mt": {
"hf": "bigscience/bloomz-7b1-mt", "hf": "bigscience/bloomz-7b1-mt",
@ -184,27 +184,27 @@ register_model_group(
models={ models={
"ChineseLLaMA2-1.3B": { "ChineseLLaMA2-1.3B": {
"hf": "hfl/chinese-llama-2-1.3b", "hf": "hfl/chinese-llama-2-1.3b",
"ms": "hfl/chinese-llama-2-1.3b", "ms": "AI-ModelScope/chinese-llama-2-1.3b",
}, },
"ChineseLLaMA2-7B": { "ChineseLLaMA2-7B": {
"hf": "hfl/chinese-llama-2-7b", "hf": "hfl/chinese-llama-2-7b",
"ms": "hfl/chinese-llama-2-7b", "ms": "AI-ModelScope/chinese-llama-2-7b",
}, },
"ChineseLLaMA2-13B": { "ChineseLLaMA2-13B": {
"hf": "hfl/chinese-llama-2-13b", "hf": "hfl/chinese-llama-2-13b",
"ms": "hfl/chinese-llama-2-13b", "ms": "AI-ModelScope/chinese-llama-2-13b",
}, },
"ChineseLLaMA2-1.3B-Chat": { "ChineseLLaMA2-1.3B-Chat": {
"hf": "hfl/chinese-alpaca-2-1.3b", "hf": "hfl/chinese-alpaca-2-1.3b",
"ms": "hfl/chinese-alpaca-2-1.3b", "ms": "AI-ModelScope/chinese-alpaca-2-1.3b",
}, },
"ChineseLLaMA2-7B-Chat": { "ChineseLLaMA2-7B-Chat": {
"hf": "hfl/chinese-alpaca-2-7b", "hf": "hfl/chinese-alpaca-2-7b",
"ms": "hfl/chinese-alpaca-2-7b", "ms": "AI-ModelScope/chinese-alpaca-2-7b",
}, },
"ChineseLLaMA2-13B-Chat": { "ChineseLLaMA2-13B-Chat": {
"hf": "hfl/chinese-alpaca-2-13b", "hf": "hfl/chinese-alpaca-2-13b",
"ms": "hfl/chinese-alpaca-2-13b", "ms": "AI-ModelScope/chinese-alpaca-2-13b",
} }
}, },
template="llama2_zh" template="llama2_zh"
@ -215,15 +215,15 @@ register_model_group(
models={ models={
"Falcon-7B": { "Falcon-7B": {
"hf": "tiiuae/falcon-7b", "hf": "tiiuae/falcon-7b",
"ms": "tiiuae/falcon-7b", "ms": "AI-ModelScope/falcon-7b",
}, },
"Falcon-40B": { "Falcon-40B": {
"hf": "tiiuae/falcon-40b", "hf": "tiiuae/falcon-40b",
"ms": "tiiuae/falcon-40b", "ms": "AI-ModelScope/falcon-40b",
}, },
"Falcon-180B": { "Falcon-180B": {
"hf": "tiiuae/falcon-180B", "hf": "tiiuae/falcon-180B",
"ms": "tiiuae/falcon-180B", "ms": "AI-ModelScope/falcon-180B",
}, },
"Falcon-7B-Chat": { "Falcon-7B-Chat": {
"hf": "tiiuae/falcon-7b-instruct", "hf": "tiiuae/falcon-7b-instruct",
@ -231,11 +231,11 @@ register_model_group(
}, },
"Falcon-40B-Chat": { "Falcon-40B-Chat": {
"hf": "tiiuae/falcon-40b-instruct", "hf": "tiiuae/falcon-40b-instruct",
"ms": "tiiuae/falcon-40b-instruct", "ms": "AI-ModelScope/falcon-40b-instruct",
}, },
"Falcon-180B-Chat": { "Falcon-180B-Chat": {
"hf": "tiiuae/falcon-180B-chat", "hf": "tiiuae/falcon-180B-chat",
"ms": "tiiuae/falcon-180B-chat", "ms": "AI-ModelScope/falcon-180B-chat",
} }
}, },
module="query_key_value", module="query_key_value",
@ -349,7 +349,7 @@ register_model_group(
models={ models={
"OpenChat3.5-7B-Chat": { "OpenChat3.5-7B-Chat": {
"hf": "openchat/openchat_3.5", "hf": "openchat/openchat_3.5",
"ms": "openchat/openchat_3.5", "ms": "myxiongmodel/openchat_3.5",
} }
}, },
template="openchat" template="openchat"
@ -360,7 +360,7 @@ register_model_group(
models={ models={
"Phi1.5-1.3B": { "Phi1.5-1.3B": {
"hf": "microsoft/phi-1_5", "hf": "microsoft/phi-1_5",
"ms": "microsoft/phi-1_5", "ms": "allspace/PHI_1-5",
} }
}, },
module="Wqkv" module="Wqkv"
@ -425,7 +425,7 @@ register_model_group(
}, },
"Vicuna1.5-13B-Chat": { "Vicuna1.5-13B-Chat": {
"hf": "lmsys/vicuna-13b-v1.5", "hf": "lmsys/vicuna-13b-v1.5",
"ms": "lmsys/vicuna-13b-v1.5", "ms": "Xorbits/vicuna-13b-v1.5",
} }
}, },
template="vicuna" template="vicuna"
@ -463,11 +463,11 @@ register_model_group(
models={ models={
"Yayi-7B": { "Yayi-7B": {
"hf": "wenge-research/yayi-7b-llama2", "hf": "wenge-research/yayi-7b-llama2",
"ms": "wenge-research/yayi-7b-llama2", "ms": "AI-ModelScope/yayi-7b-llama2",
}, },
"Yayi-13B": { "Yayi-13B": {
"hf": "wenge-research/yayi-13b-llama2", "hf": "wenge-research/yayi-13b-llama2",
"ms": "wenge-research/yayi-13b-llama2", "ms": "AI-ModelScope/yayi-13b-llama2",
} }
}, },
template="yayi" template="yayi"
@ -501,7 +501,7 @@ register_model_group(
models={ models={
"Zephyr-7B-Alpha-Chat": { "Zephyr-7B-Alpha-Chat": {
"hf": "HuggingFaceH4/zephyr-7b-alpha", "hf": "HuggingFaceH4/zephyr-7b-alpha",
"ms": "HuggingFaceH4/zephyr-7b-alpha", "ms": "AI-ModelScope/zephyr-7b-alpha",
}, },
"Zephyr-7B-Beta-Chat": { "Zephyr-7B-Beta-Chat": {
"hf": "HuggingFaceH4/zephyr-7b-beta", "hf": "HuggingFaceH4/zephyr-7b-beta",