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

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

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
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 1
      colossalai/checkpoint_io/utils.py

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