423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-14 10:56:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

add dataset stage and filter dataset when stage chosen in webui

Browse Source 
Former-commit-id: 26e4136449a4df6028d834fd16a0f4a7c532759d
This commit is contained in:
codemayq
2023-08-23 18:54:23 +08:00
parent 312b9513ef
commit 22cece8acb
4 changed files with 84 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

99
data/dataset_info.json
View File

@@ -1,23 +1,28 @@
{
"alpaca_en": {
"file_name": "alpaca_data_en_52k.json",
"file_sha1": "607f94a7f581341e59685aef32f531095232cf23"
"file_sha1": "607f94a7f581341e59685aef32f531095232cf23",
"stage": "sft"
},
"alpaca_zh": {
"file_name": "alpaca_data_zh_51k.json",
"file_sha1": "e655af3db557a4197f7b0cf92e1986b08fae6311"
"file_sha1": "e655af3db557a4197f7b0cf92e1986b08fae6311",
"stage": "sft"
},
"alpaca_gpt4_en": {
"file_name": "alpaca_gpt4_data_en.json",
"file_sha1": "647f4ad447bd993e4b6b6223d1be15208bab694a"
"file_sha1": "647f4ad447bd993e4b6b6223d1be15208bab694a",
"stage": "sft"
},
"alpaca_gpt4_zh": {
"file_name": "alpaca_gpt4_data_zh.json",
"file_sha1": "3eaa3bda364ccdd59925d7448a698256c31ef845"
"file_sha1": "3eaa3bda364ccdd59925d7448a698256c31ef845",
"stage": "sft"
},
"self_cognition": {
"file_name": "self_cognition.json",
"file_sha1": "6287a730ada924fc5d9eadc6d8f865e01b7a6f67"
"file_sha1": "6287a730ada924fc5d9eadc6d8f865e01b7a6f67",
"stage": "sft"
},
"oaast_sft": {
"file_name": "oaast_sft.json",
@@ -27,7 +32,8 @@
"query": "input",
"response": "output",
"history": "history"
}
},
"stage": "sft"
},
"oaast_sft_zh": {
"file_name": "oaast_sft_zh.json",
@@ -37,7 +43,8 @@
"query": "input",
"response": "output",
"history": "history"
}
},
"stage": "sft"
},
"sharegpt_zh": {
"file_name": "sharegpt_zh_27k.json",
@@ -47,7 +54,8 @@
"query": "input",
"response": "output",
"history": "history"
}
},
"stage": "sft"
},
"lima": {
"file_name": "lima.json",
@@ -57,7 +65,8 @@
"query": "input",
"response": "output",
"history": "history"
}
},
"stage": "sft"
},
"example": {
"script_url": "example_dataset",
@@ -66,25 +75,32 @@
"query": "input",
"response": "output",
"history": "history"
}
},
"stage": "sft"
},
"guanaco": {
"hf_hub_url": "JosephusCheung/GuanacoDataset"
"hf_hub_url": "JosephusCheung/GuanacoDataset",
"stage": "sft"
},
"belle_0.5m": {
"hf_hub_url": "BelleGroup/train_0.5M_CN"
"hf_hub_url": "BelleGroup/train_0.5M_CN",
"stage": "sft"
},
"belle_1m": {
"hf_hub_url": "BelleGroup/train_1M_CN"
"hf_hub_url": "BelleGroup/train_1M_CN",
"stage": "sft"
},
"belle_2m": {
"hf_hub_url": "BelleGroup/train_2M_CN"
"hf_hub_url": "BelleGroup/train_2M_CN",
"stage": "sft"
},
"belle_dialog": {
"hf_hub_url": "BelleGroup/generated_chat_0.4M"
"hf_hub_url": "BelleGroup/generated_chat_0.4M",
"stage": "sft"
},
"belle_math": {
"hf_hub_url": "BelleGroup/school_math_0.25M"
"hf_hub_url": "BelleGroup/school_math_0.25M",
"stage": "sft"
},
"belle_multiturn": {
"script_url": "belle_multiturn",
@@ -93,7 +109,8 @@
"query": "",
"response": "output",
"history": "history"
}
},
"stage": "sft"
},
"firefly": {
"hf_hub_url": "YeungNLP/firefly-train-1.1M",
@@ -102,13 +119,16 @@
"query": "",
"response": "target",
"history": ""
}
},
"stage": "sft"
},
"codealpaca": {
"hf_hub_url": "sahil2801/CodeAlpaca-20k"
"hf_hub_url": "sahil2801/CodeAlpaca-20k",
"stage": "sft"
},
"alpaca_cot": {
"hf_hub_url": "QingyiSi/Alpaca-CoT"
"hf_hub_url": "QingyiSi/Alpaca-CoT",
"stage": "sft"
},
"webqa": {
"hf_hub_url": "suolyer/webqa",
@@ -117,7 +137,8 @@
"query": "",
"response": "output",
"history": ""
}
},
"stage": "sft"
},
"ultra_chat": {
"script_url": "ultra_chat",
@@ -126,18 +147,22 @@
"query": "",
"response": "output",
"history": "history"
}
},
"stage": "sft"
},
"novel_tokens512_50k": {
"hf_hub_url": "zxbsmk/webnovel_cn"
"hf_hub_url": "zxbsmk/webnovel_cn",
"stage": "sft"
},
"comparison_gpt4_en": {
"file_name": "comparison_gpt4_data_en.json",
"file_sha1": "96fa18313544e22444fe20eead7754b17da452ae"
"file_sha1": "96fa18313544e22444fe20eead7754b17da452ae",
"stage": "rm"
},
"comparison_gpt4_zh": {
"file_name": "comparison_gpt4_data_zh.json",
"file_sha1": "515b18ed497199131ddcc1af950345c11dc5c7fd"
"file_sha1": "515b18ed497199131ddcc1af950345c11dc5c7fd",
"stage": "rm"
},
"hh_rlhf_en": {
"script_url": "hh_rlhf_en",
@@ -146,7 +171,8 @@
"query": "",
"response": "output",
"history": "history"
}
},
"stage": "rm"
},
"oaast_rm": {
"file_name": "oaast_rm.json",
@@ -156,7 +182,8 @@
"query": "input",
"response": "output",
"history": "history"
}
},
"stage": "rm"
},
"oaast_rm_zh": {
"file_name": "oaast_rm_zh.json",
@@ -166,7 +193,8 @@
"query": "input",
"response": "output",
"history": "history"
}
},
"stage": "rm"
},
"wiki_demo": {
"file_name": "wiki_demo.txt",
@@ -176,7 +204,8 @@
"query": "",
"response": "",
"history": ""
}
},
"stage": "pt"
},
"refinedweb": {
"hf_hub_url": "tiiuae/falcon-refinedweb",
@@ -185,7 +214,8 @@
"query": "",
"response": "",
"history": ""
}
},
"stage": "pt"
},
"starcoder": {
"hf_hub_url": "bigcode/starcoderdata",
@@ -194,7 +224,8 @@
"query": "",
"response": "",
"history": ""
}
},
"stage": "pt"
},
"wikipedia_en": {
"hf_hub_url": "olm/olm-wikipedia-20221220",
@@ -203,7 +234,8 @@
"query": "",
"response": "",
"history": ""
}
},
"stage": "pt"
},
"wikipedia_zh": {
"hf_hub_url": "pleisto/wikipedia-cn-20230720-filtered",
@@ -212,6 +244,7 @@
"query": "",
"response": "",
"history": ""
}
},
"stage": "pt"
}
}