for test

Former-commit-id: 33d9082320098f994bfa0c6353459afcb93165b7
2023-12-12 11:52:59 +08:00
parent b634e91c43
commit 79a376d1db
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,