ColossalAI/tests/test_booster/test_plugin
Frank Lee 1beb85cc25
[checkpoint] refactored the API and added safetensors support (#3427)
* [checkpoint] refactored the API and added safetensors support

* polish code
2023-04-04 15:23:01 +08:00
..
test_gemini_plugin.py [test] fixed gemini plugin test (#3411) 2023-04-03 17:12:22 +08:00
test_torch_ddp_plugin.py [checkpoint] refactored the API and added safetensors support (#3427) 2023-04-04 15:23:01 +08:00