tests/test_checkpoint_io/test_gemini_checkpoint_io.py · 41e52c1c6eaf1d0683427aa59bdd777ae1458bd7 · OpenDAS / ColossalAI

flybird11111 authored Nov 16, 2023

* support ddp

* fix

* fix

* fix

fix

* support ddp

* fix

* fix

* fix

fix

* simplify tests

* fix

* fix

* fix

fix

fix

* fix

3e021547

test_gemini_checkpoint_io.py 6.93 KB