From 4603538dddc7957bc3ebc29caa066471da2417ba Mon Sep 17 00:00:00 2001 From: Ziyue Jiang Date: Wed, 15 Feb 2023 10:53:38 +0800 Subject: [PATCH] [NFC] posh colossalai/context/process_group_initializer/initializer_sequence.py code style (#2712) Co-authored-by: Ziyue Jiang --- .../context/process_group_initializer/initializer_sequence.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/colossalai/context/process_group_initializer/initializer_sequence.py b/colossalai/context/process_group_initializer/initializer_sequence.py index 682fe4bb7..eaacb14d2 100644 --- a/colossalai/context/process_group_initializer/initializer_sequence.py +++ b/colossalai/context/process_group_initializer/initializer_sequence.py @@ -3,9 +3,10 @@ import torch.distributed as dist from colossalai.registry import DIST_GROUP_INITIALIZER + +from ..parallel_mode import ParallelMode from .initializer_tensor import Initializer_Tensor from .process_group_initializer import ProcessGroupInitializer -from ..parallel_mode import ParallelMode @DIST_GROUP_INITIALIZER.register_module