mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-04-12 15:14:55 +00:00
d6abd933f2
* 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']
|