423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-04 12:42:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix #1290

Browse Source 
Former-commit-id: eb9d9e104ad5cdfc23f9cc48f20068c674762c47
This commit is contained in:
hiyouga 2023-11-03 00:44:53 +08:00
parent 5507014392
commit 574a7d175c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/llmtuner/tuner/core/parser.py
View File

@ -106,8 +106,8 @@ def get_train_args(
if finetuning_args.stage in ["rm", "ppo"] and training_args.resume_from_checkpoint is not None: if finetuning_args.stage in ["rm", "ppo"] and training_args.resume_from_checkpoint is not None:
raise ValueError("RM and PPO stages do not support `resume_from_checkpoint`.") raise ValueError("RM and PPO stages do not support `resume_from_checkpoint`.")
if finetuning_args.stage in ["ppo", "dpo"] and not training_args.do_train: if finetuning_args.stage == "ppo" and not training_args.do_train:
raise ValueError("PPO and DPO stages can only be performed at training.") raise ValueError("PPO training does not support evaluation.")
if finetuning_args.stage in ["rm", "dpo"]: if finetuning_args.stage in ["rm", "dpo"]:
for dataset_attr in data_args.dataset_list: for dataset_attr in data_args.dataset_list: