mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-10-22 07:14:09 +00:00
de0468c7a8
* add zero init context * add more flags for zero init context fix bug of repeated converting param to ShardedParamV2 * polish code
4 lines
73 B
Python
4 lines
73 B
Python
from .init_context import ZeroInitContext
|
|
|
|
__all__ = ['ZeroInitContext']
|