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

fix data loader hint

Browse Source 
Former-commit-id: 25b56126a11591b0155e2f72b673dd8f45a6c8c9
This commit is contained in:
hiyouga 2024-06-03 18:28:27 +08:00
parent 070b944895
commit 123a845209
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/llamafactory/data/loader.py
View File

@ -62,9 +62,9 @@ def load_single_dataset(
raise ValueError("File {} not found.".format(local_path)) raise ValueError("File {} not found.".format(local_path))
if data_path is None: if data_path is None:
raise ValueError("File extension must be txt, csv, json or jsonl.") raise ValueError("Allowed file types: {}.".format(",".join(FILEEXT2TYPE.keys())))
else: else:
raise NotImplementedError raise NotImplementedError("Unknown load type: {}.".format(dataset_attr.load_from))
if dataset_attr.load_from == "ms_hub": if dataset_attr.load_from == "ms_hub":
try: try: