update dc link

Former-commit-id: f6789e50e17a377b6d9b434d8e12ad99d8eecfeb
This commit is contained in:
hiyouga
2023-12-15 22:11:31 +08:00
parent d8d74091f6
commit 9121722999
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