Skip to content

GitLab

Unverified Commit 7e46bc87 authored by digger yu's avatar digger yu Committed by GitHub
Browse files

fix CheckpointIndexFile is not defined (#4109)

parent 09fe9dc7
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
colossalai/checkpoint_io/utils.py
View file @ 7e46bc87
......@@ -11,6 +11,7 @@ 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"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment