fix #1218

Former-commit-id: b301f35bd4a3bf368159c8f5fb4e2736f922115b
2023-10-19 16:17:41 +08:00
parent 142dd63b47
commit 47a1f73d0f
3 changed files with 11 additions and 6 deletions
--- a/src/llmtuner/tuner/core/parser.py
+++ b/src/llmtuner/tuner/core/parser.py
@@ -88,8 +88,8 @@ def get_train_args(
    transformers.utils.logging.enable_default_handler()
    transformers.utils.logging.enable_explicit_format()

-    # Check arguments (do not check finetuning_args since it may be loaded from checkpoints)
-    data_args.init_for_training()
+    # Check arguments
+    data_args.init_for_training(training_args.seed)

    if finetuning_args.stage != "pt" and data_args.template is None:
        raise ValueError("Please specify which `template` to use.")