SIGN IN SIGN UP
hpcaitech / ColossalAI UNCLAIMED

Making large AI models cheaper, faster and more accessible

0 0 37 Python

[checkpointio] fix gemini and hybrid parallel optim checkpoint (#5347)

* [checkpointio] fix hybrid parallel optim checkpoint

* [extension] fix cuda extension

* [checkpointio] fix gemini optimizer checkpoint

* polish code
H
Hongxin Liu committed
ffffc32dc71c767aac625279593a7ab339855645
Parent: c523984
Committed by GitHub <noreply@github.com> on 2/1/2024, 8:13:06 AM