mirror of https://github.com/hpcaitech/ColossalAI
1beb85cc25
* [checkpoint] refactored the API and added safetensors support * polish code |
||
---|---|---|
.. | ||
test_gemini_plugin.py | ||
test_torch_ddp_plugin.py |