mirror of https://github.com/hpcaitech/ColossalAI
![]() * add load_state_dict for sharded model * fix bug * fix bug * fix ckpt dtype and device * support load state dict in zero init ctx * fix bugs |
||
---|---|---|
.. | ||
__init__.py | ||
init_context.py |