mirror of https://github.com/hpcaitech/ColossalAI
![]() * [gemini] save state dict support fp16 * [gemini] save state dict shard support fp16 * [gemini] fix state dict * [gemini] fix state dict |
||
---|---|---|
.. | ||
gemini | ||
legacy | ||
low_level | ||
__init__.py | ||
wrapper.py |