Unverified Commit 4603538d authored by Ziyue Jiang's avatar Ziyue Jiang Committed by GitHub
Browse files

[NFC] posh colossalai/context/process_group_initializer/initializer_sequence.py code style (#2712)


Co-authored-by: default avatarZiyue Jiang <ziyue.jiang@gmail.com>
parent b3d10db5
...@@ -3,9 +3,10 @@ ...@@ -3,9 +3,10 @@
import torch.distributed as dist import torch.distributed as dist
from colossalai.registry import DIST_GROUP_INITIALIZER from colossalai.registry import DIST_GROUP_INITIALIZER
from ..parallel_mode import ParallelMode
from .initializer_tensor import Initializer_Tensor from .initializer_tensor import Initializer_Tensor
from .process_group_initializer import ProcessGroupInitializer from .process_group_initializer import ProcessGroupInitializer
from ..parallel_mode import ParallelMode
@DIST_GROUP_INITIALIZER.register_module @DIST_GROUP_INITIALIZER.register_module
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment