fix #2397

Former-commit-id: 7404692808f2288d539668d364965ad104dacadb
2024-02-03 23:45:31 +08:00
parent a2ae5bd867
commit 996cc5d900
1 changed files with 0 additions and 3 deletions
--- a/src/llmtuner/data/loader.py
+++ b/src/llmtuner/data/loader.py
@@ -155,9 +155,6 @@ def get_dataset(
                dataset = dataset.to_iterable_dataset()
            return dataset

-        if data_args.streaming:
-            raise ValueError("Turn off dataset streaming to save cache files.")
-
    with training_args.main_process_first(desc="load dataset"):
        all_datasets = []
        for dataset_attr in get_dataset_list(data_args):  # TODO: add split