From 1350ece492f47f769ccc871fc79bc780b333b351 Mon Sep 17 00:00:00 2001
From: Baizhou Zhang <eddiezhang@pku.edu.cn>
Date: Mon, 3 Jul 2023 22:14:37 +0800
Subject: [PATCH] [hotfix] fix import bug in checkpoint_io (#4142)

---
 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 f28b3e82c..3dada00cd 100644
--- a/colossalai/checkpoint_io/utils.py
+++ b/colossalai/checkpoint_io/utils.py
@@ -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"