github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

bug

pull/529/head
lijiaxing 2023-12-12 10:36:04 +08:00
parent 66e4a8a847
commit d6eeacfeb2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internlm/utils/model_checkpoint.py
View File

@ -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():