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

from .init_context import ZeroInitContext, no_shard_zero_context, no_shard_zero_decrator

__all__ = ["ZeroInitContext", "no_shard_zero_context", "no_shard_zero_decrator"]
[zero] add zero context manager to change config during initialization (#546) 2022-03-29 09:57:59 +00:00			`from .init_context import ZeroInitContext, no_shard_zero_context, no_shard_zero_decrator`
[zero] zero init context (#321) * add zero init context * add more flags for zero init context fix bug of repeated converting param to ShardedParamV2 * polish code 2022-03-07 08:14:40 +00:00
[misc] update pre-commit and run all files (#4752) * [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format 2023-09-19 06:20:26 +00:00			`__all__ = ["ZeroInitContext", "no_shard_zero_context", "no_shard_zero_decrator"]`