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

[hotfix] fix import bug in checkpoint_io (#4142)

Browse Source
pull/4157/head
Baizhou Zhang 1 year ago committed by GitHub
parent 8abc87798f
commit 1350ece492
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File

1
colossalai/checkpoint_io/utils.py
Unescape View File

@ -11,7 +11,6 @@ import torch.nn as nn
from torch.optim import Optimizer
from colossalai.tensor.d_tensor.d_tensor import DTensor
from .index_file import CheckpointIndexFile
SAFE_WEIGHTS_NAME = "model.safetensors"
WEIGHTS_NAME = "pytorch_model.bin"

Write Preview
Loading…
Cancel
Save