update dc link

Former-commit-id: 87ef3f47b5
This commit is contained in:
hiyouga
2023-12-15 22:11:31 +08:00
parent 68b1500c41
commit 2db4cfab40
3 changed files with 3 additions and 2 deletions

View File

@@ -145,6 +145,7 @@ def get_train_args(args: Optional[Dict[str, Any]] = None) -> _TRAIN_CLS:
if finetuning_args.stage in ["rm", "ppo"] and finetuning_args.finetuning_type in ["full", "freeze"]:
can_resume_from_checkpoint = False
training_args.resume_from_checkpoint = None
else:
can_resume_from_checkpoint = True