ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix constants

Browse Source 
Former-commit-id: e66a338410be6812064a119d8c6a6644e0f035d1
This commit is contained in:
BUAADreamer
2024-09-29 22:40:43 +08:00
parent 28d09b81c9
commit a96c5da737
2 changed files with 44 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
src/llamafactory/extras/constants.py
View File

@@ -854,12 +854,6 @@ register_model_group(
"LLaVA-NeXT-LLaMA3-8B-Chat": {
DownloadSource.DEFAULT: "llava-hf/llama3-llava-next-8b-hf",
},
"LLaVA-NeXT-LLaMA3-72B-Chat": {
DownloadSource.DEFAULT: "llava-hf/llava-next-72b-hf",
},
"LLaVA-NeXT-LLaMA3-110B-Chat": {
DownloadSource.DEFAULT: "llava-hf/llava-next-110b-hf",
},
},
template="llava_next_llama3",
vision=True,
@@ -877,6 +871,20 @@ register_model_group(
)
register_model_group(
models={
"LLaVA-NeXT-Qwen-1.5-72B-Chat": {
DownloadSource.DEFAULT: "llava-hf/llava-next-72b-hf",
},
"LLaVA-NeXT-Qwen-1.5-110B-Chat": {
DownloadSource.DEFAULT: "llava-hf/llava-next-110b-hf",
},
},
template="llava_next_qwen",
vision=True,
)
register_model_group(
models={
"LLaVA-NeXT-Yi-34B-Chat": {
@@ -893,9 +901,6 @@ register_model_group(
"LLaVA-NeXT-Video-7B-Chat": {
DownloadSource.DEFAULT: "llava-hf/LLaVA-NeXT-Video-7B-hf",
},
"LLaVA-NeXT-Video-7B-32k-Chat": {
DownloadSource.DEFAULT: "llava-hf/LLaVA-NeXT-Video-7B-32K-hf",
},
"LLaVA-NeXT-Video-7B-DPO-Chat": {
DownloadSource.DEFAULT: "llava-hf/LLaVA-NeXT-Video-7B-DPO-hf",
},
@@ -905,6 +910,17 @@ register_model_group(
)
register_model_group(
models={
"LLaVA-NeXT-Video-Mistral-7B-32k-Chat": {
DownloadSource.DEFAULT: "llava-hf/LLaVA-NeXT-Video-7B-32K-hf",
},
},
template="llava_next_video_mistral",
vision=True,
)
register_model_group(
models={
"LLaVA-NeXT-Video-Yi-34B-Chat": {