From a4633cc12f4ff9e06de7213323240aaeced8248c Mon Sep 17 00:00:00 2001 From: wql Date: Thu, 26 Sep 2024 14:09:27 +0800 Subject: [PATCH] test: retry --- src/llamafactory/data/loader.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/llamafactory/data/loader.py b/src/llamafactory/data/loader.py index 55b3ab91..720fa493 100644 --- a/src/llamafactory/data/loader.py +++ b/src/llamafactory/data/loader.py @@ -213,9 +213,9 @@ def get_dataset( if has_tokenized_data(data_args.tokenized_path): logger.warning("Loading dataset from disk will ignore other data arguments.") dataset_dict: "DatasetDict" = load_from_disk(data_args.tokenized_path) - # print(data_args.tokenized_path) - # print(dataset_dict) - # time.sleep(100) + print(data_args.tokenized_path) + print(dataset_dict) + time.sleep(100) logger.info("Loaded tokenized dataset from {}.".format(data_args.tokenized_path))