From 29f6fb5a5d90714abd0e000bbda5b22dc4b339e6 Mon Sep 17 00:00:00 2001 From: digger yu Date: Wed, 21 Jun 2023 15:47:56 +0800 Subject: [PATCH] fix #3852 path error --- examples/language/gpt/titans/train_gpt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/language/gpt/titans/train_gpt.py b/examples/language/gpt/titans/train_gpt.py index 66225d6c8044..6be0b9e8da30 100644 --- a/examples/language/gpt/titans/train_gpt.py +++ b/examples/language/gpt/titans/train_gpt.py @@ -15,7 +15,7 @@ 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):