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

tiny fix

Browse Source
This commit is contained in:
hiyouga committed 2024-06-04 00:31:10 +08:00
1 parent 91611d68c4
commit 5a13b3baa6
3 files changed
+5 -9
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/llamafactory/data/loader.py
+1 -1
View File
@@ -179,7 +179,7 @@ def get_dataset(
if training_args.should_save:
dataset.save_to_disk(data_args.tokenized_path)
logger.info("Tokenized dataset saved at {}.".format(data_args.tokenized_path))
logger.info("Please restart the training with `--tokenized_path {}`.".format(data_args.tokenized_path))
logger.info("Please restart the training with `tokenized_path: {}`.".format(data_args.tokenized_path))
sys.exit(0)