diff --git a/train_sft_ds.py b/train_sft_ds.py index a3eaa76..10024fa 100644 --- a/train_sft_ds.py +++ b/train_sft_ds.py @@ -216,6 +216,8 @@ def parse_args(): ap.add_argument("--wandb_project", type=str, default="ds-qwen3") ap.add_argument("--eval_data_glob", type=str, default=None, help="(可选) 测试/验证集 jsonl 通配符;如提供则优先使用") + ap.add_argument("--local_rank", type=int, default=-1, + help="for deepspeed/torchrun launcher; ignored by user code") return ap.parse_args()