for test

2023-12-12 11:52:59 +08:00 · 2023-12-12 11:52:59 +08:00 · 5b979147f0
parent 8a908a8c64
commit 5b979147f0
1 changed files with 1 additions and 1 deletions
--- a/src/llmtuner/data/loader.py
+++ b/src/llmtuner/data/loader.py
@ -65,7 +65,7 @@ def get_dataset(
                example['input'] = example['input'] or ''
                return example
-            dataset = dataset.ds_instance.map(map_func)
+            dataset = dataset.map(map_func)
        else:
            dataset = load_dataset(
                path=data_path,