diff --git a/internlm/utils/model_checkpoint.py b/internlm/utils/model_checkpoint.py index dfd0386..ee6266d 100644 --- a/internlm/utils/model_checkpoint.py +++ b/internlm/utils/model_checkpoint.py @@ -910,7 +910,7 @@ now step_count is {train_state.step_count}", f"Unsupported backend: {backend}, " "Currently only support `boto3`, `oss2`, `volc` and `local`" ) - if latest_ckpt and not latest_ckpt.startswith(backend): + if latest_ckpt and not latest_ckpt.startswith(backend + ":"): latest_ckpt = ":".join([backend, latest_ckpt]) if gpc.is_rank_for_log():