fix #3852 path error (#4058)

2d40759a · digger yu · GitHub · 1ee947f6 · 2d40759a
Unverified Commit 2d40759a authored Jun 28, 2023 by digger yu Committed by GitHub Jun 28, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/language/gpt/titans/train_gpt.py examples/language/gpt/titans/train_gpt.py +1 -1

No files found.
--- a/examples/language/gpt/titans/train_gpt.py
+++ b/examples/language/gpt/titans/train_gpt.py
@@ -15,7 +15,7 @@ from colossalai.nn import LinearWarmupLR
 from colossalai.trainer import Trainer, hooks
 from colossalai.utils import colo_set_process_memory_fraction, is_using_pp
 from colossalai.utils.timer import MultiTimer
-from colossalai.zero.init_ctx import ZeroInitContext
+from colossalai.zero.legacy.init_ctx import ZeroInitContext


 def calc_local_model_size(model: torch.nn.Module):