mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-03 04:02:49 +08:00
fix dataset
Former-commit-id: 046b6fb118e3ea75062c6a759720a1759639e93c
This commit is contained in:
parent
d51d96d594
commit
9822cb7bac
@ -297,7 +297,8 @@
|
|||||||
}
|
}
|
||||||
},
|
},
|
||||||
"openo1_sft": {
|
"openo1_sft": {
|
||||||
"hf_hub_url": "O1-OPEN/OpenO1-SFT",
|
"hf_hub_url": "llamafactory/OpenO1-SFT",
|
||||||
|
"ms_hub_url": "llamafactory/OpenO1-SFT",
|
||||||
"columns": {
|
"columns": {
|
||||||
"prompt": "prompt",
|
"prompt": "prompt",
|
||||||
"response": "response"
|
"response": "response"
|
||||||
|
@ -512,6 +512,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from chatml template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="chatml_de",
|
name="chatml_de",
|
||||||
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
||||||
@ -578,6 +579,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from chatml template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="dbrx",
|
name="dbrx",
|
||||||
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
||||||
@ -728,6 +730,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from llama2 template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="llama2_zh",
|
name="llama2_zh",
|
||||||
format_user=StringFormatter(slots=[{"bos_token"}, "[INST] {{content}} [/INST]"]),
|
format_user=StringFormatter(slots=[{"bos_token"}, "[INST] {{content}} [/INST]"]),
|
||||||
@ -762,6 +765,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from llama3 template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="mllama",
|
name="mllama",
|
||||||
format_user=StringFormatter(
|
format_user=StringFormatter(
|
||||||
@ -789,6 +793,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from vicuna template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="llava",
|
name="llava",
|
||||||
format_user=StringFormatter(slots=["USER: {{content}} ASSISTANT:"]),
|
format_user=StringFormatter(slots=["USER: {{content}} ASSISTANT:"]),
|
||||||
@ -800,6 +805,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from vicuna template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="llava_next",
|
name="llava_next",
|
||||||
format_user=StringFormatter(slots=["USER: {{content}} ASSISTANT:"]),
|
format_user=StringFormatter(slots=["USER: {{content}} ASSISTANT:"]),
|
||||||
@ -811,6 +817,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from llama3 template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="llava_next_llama3",
|
name="llava_next_llama3",
|
||||||
format_user=StringFormatter(
|
format_user=StringFormatter(
|
||||||
@ -838,6 +845,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from mistral template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="llava_next_mistral",
|
name="llava_next_mistral",
|
||||||
format_user=StringFormatter(slots=["[INST] {{content}} [/INST]"]),
|
format_user=StringFormatter(slots=["[INST] {{content}} [/INST]"]),
|
||||||
@ -846,6 +854,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from chatml template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="llava_next_qwen",
|
name="llava_next_qwen",
|
||||||
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
||||||
@ -860,6 +869,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from chatml template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="llava_next_yi",
|
name="llava_next_yi",
|
||||||
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
||||||
@ -871,6 +881,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from vicuna template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="llava_next_video",
|
name="llava_next_video",
|
||||||
format_user=StringFormatter(slots=["USER: {{content}} ASSISTANT:"]),
|
format_user=StringFormatter(slots=["USER: {{content}} ASSISTANT:"]),
|
||||||
@ -882,6 +893,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from mistral template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="llava_next_video_mistral",
|
name="llava_next_video_mistral",
|
||||||
format_user=StringFormatter(slots=["[INST] {{content}} [/INST]"]),
|
format_user=StringFormatter(slots=["[INST] {{content}} [/INST]"]),
|
||||||
@ -890,6 +902,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from chatml template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="llava_next_video_yi",
|
name="llava_next_video_yi",
|
||||||
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
||||||
@ -901,6 +914,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from chatml template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="marco",
|
name="marco",
|
||||||
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
||||||
@ -954,6 +968,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from chatml template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="opencoder",
|
name="opencoder",
|
||||||
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
||||||
@ -974,6 +989,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from gemma template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="paligemma",
|
name="paligemma",
|
||||||
format_user=StringFormatter(slots=["<start_of_turn>user\n{{content}}<end_of_turn>\n<start_of_turn>model\n"]),
|
format_user=StringFormatter(slots=["<start_of_turn>user\n{{content}}<end_of_turn>\n<start_of_turn>model\n"]),
|
||||||
@ -1017,6 +1033,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from chatml template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="qwen",
|
name="qwen",
|
||||||
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
||||||
@ -1030,6 +1047,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from chatml template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="qwen2_vl",
|
name="qwen2_vl",
|
||||||
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
||||||
@ -1058,6 +1076,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from llama3 template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="skywork_o1",
|
name="skywork_o1",
|
||||||
format_user=StringFormatter(
|
format_user=StringFormatter(
|
||||||
@ -1175,6 +1194,7 @@ _register_template(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# copied from chatml template
|
||||||
_register_template(
|
_register_template(
|
||||||
name="yi",
|
name="yi",
|
||||||
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
|
||||||
|
Loading…
x
Reference in New Issue
Block a user