ColossalAI/colossalai/legacy/zero/init_ctx/__init__.py

mirror of https://github.com/hpcaitech/ColossalAI
from .init_context import ZeroInitContext, no_shard_zero_context, no_shard_zero_decrator

__all__ = ["ZeroInitContext", "no_shard_zero_context", "no_shard_zero_decrator"]
	`from .init_context import ZeroInitContext, no_shard_zero_context, no_shard_zero_decrator`

	`__all__ = ["ZeroInitContext", "no_shard_zero_context", "no_shard_zero_decrator"]`