Skip to content

GitLab

Unverified Commit 1350ece4 authored by Baizhou Zhang's avatar Baizhou Zhang Committed by GitHub
Browse files

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

parent 8abc8779
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
colossalai/checkpoint_io/utils.py
View file @ 1350ece4
...@@ -11,7 +11,6 @@ import torch.nn as nn ...@@ -11,7 +11,6 @@ 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"
......
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