Modify the design

hpcaitech · Jan 20, 2025 · 17ee5a7 · 17ee5a7
1 parent c34ba4e
commit 17ee5a7
Show file tree

Hide file tree

Showing 6 changed files with 49 additions and 647 deletions.
diff --git a/colossalai/checkpoint_io/__init__.py b/colossalai/checkpoint_io/__init__.py
@@ -2,7 +2,6 @@
 from .general_checkpoint_io import GeneralCheckpointIO
 from .hybrid_parallel_checkpoint_io import HybridParallelCheckpointIO
 
-from .distributed_checkpoint_io import DistributedCheckpointIO
 from .index_file import CheckpointIndexFile
 from .moe_checkpoint import MoECheckpointIO
 
@@ -12,5 +11,4 @@
     "GeneralCheckpointIO",
     "HybridParallelCheckpointIO",
     "MoECheckpointIO",
-    "DistributedCheckpointIO",
 ]