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

fix CheckpointIndexFile is not defined (#4109)

pull/4142/head
digger yu 2023-07-03 17:09:06 +08:00 committed by GitHub
parent 09fe9dc704
commit 7e46bc87b6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
colossalai/checkpoint_io/utils.py
View File

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