kyy/llm_trainer
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

for test

Browse Source
This commit is contained in:
xingjun.wang committed 2023-12-12 11:52:59 +08:00
1 parent 8a908a8c64
commit 5b979147f0
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
src/llmtuner/data/loader.py
+1 -1
View File
@@ -65,7 +65,7 @@ def get_dataset(
example['input'] = example['input'] or '' example['input'] = example['input'] or ''
return example return example
dataset = dataset.ds_instance.map(map_func) dataset = dataset.map(map_func)
else: else:
dataset = load_dataset( dataset = load_dataset(
path=data_path, path=data_path,