423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-22 05:32:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix dataset

Browse Source 
Former-commit-id: 487dee066f51caea86dc65847bd326e13445de09
This commit is contained in:
hiyouga 2024-01-18 12:59:30 +08:00
parent 7f12aedc08
commit e95c0242a8
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
data/glaive_toolcall_10k.json.REMOVED.git-id
View File

@ -1 +1 @@
0fe460c57c1a1dae70cdd72b185129c2aaae4e24
4748dff00d1dc42768a5b6cc772143c313017812

2
src/llmtuner/data/aligner.py
View File

@ -60,7 +60,7 @@ def convert_sharegpt(examples: Dict[str, List[Any]], dataset_attr: "DatasetAttr"
accept_tags = [dataset_attr.assistant_tag, dataset_attr.function_tag]
if message[dataset_attr.role_tag] not in accept_tags:
raise ValueError("Invalid role tag.")
raise ValueError("Invalid role tag in {}.".format(messages))
prompt.append({"role": tag_mapping[message[dataset_attr.role_tag]], "content": message[dataset_attr.content_tag]})