fix: 修复特殊token 问题

This commit is contained in:
2026-04-02 16:16:02 +08:00
parent e01ec081b3
commit f44ad6912e
28 changed files with 334 additions and 163 deletions
@@ -7,14 +7,28 @@ normalizer = TextNormalizer()
def process_func(input_dict: dict):
query = input_dict["prompt"] if input_dict["prompt"] else ""
resp = input_dict["response"] if input_dict["response"] else ""
return {"query": normalizer.normalize(query), "response": normalizer.normalize(resp)}
return {
"query": normalizer.normalize(query),
"response": normalizer.normalize(resp),
}
if __name__ == "__main__":
all_data = [
'stem_zh', 'infinity-instruct', 'firefly', 'magpie', 'dpsk-r1-distil',
'coig-cqia', 'disc-law', 'neo_sft_phase2', 'chinese-medical', 'chinese-reasoning-distil',
'psycho-10k-dpsk-r1', 'sof-c-zh', 'industryinstruction', 'Chinese-QA-AFAF',
"stem_zh",
"infinity-instruct",
"firefly",
"magpie",
"dpsk-r1-distil",
"coig-cqia",
"disc-law",
"neo_sft_phase2",
"chinese-medical",
"chinese-reasoning-distil",
"psycho-10k-dpsk-r1",
"sof-c-zh",
"industryinstruction",
"Chinese-QA-AFAF",
]
dataset_list = []