From 01266239cfbb1fec02adce464c2821dbd210b1a3 Mon Sep 17 00:00:00 2001 From: Baizhou Zhang Date: Mon, 3 Jul 2023 17:30:15 +0800 Subject: [PATCH] [hotfix] fix import bug in checkpoint_io --- colossalai/checkpoint_io/utils.py | 1 - 1 file changed, 1 deletion(-) diff --git a/colossalai/checkpoint_io/utils.py b/colossalai/checkpoint_io/utils.py index f28b3e82cc9e..3dada00cd9b5 100644 --- a/colossalai/checkpoint_io/utils.py +++ b/colossalai/checkpoint_io/utils.py @@ -11,7 +11,6 @@ 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"