diff --git a/README.md b/README.md index 88bb5b06b..baff74a49 100644 --- a/README.md +++ b/README.md @@ -113,7 +113,7 @@ Project stages: - [x] Support mixed precision training. - [x] Add gradient checkpoint. - [x] Support for ReBased and Ring attention. ๐Ÿค Thanks to [@kabachuha](https://github.com/kabachuha) - - [ ] Train using the deepspeed engine. ๐Ÿ™ **[Need your contribution]** + - [x] Train using the deepspeed engine. ๐Ÿค Thanks to [@sennnnn](https://github.com/sennnnn๏ผ‰ - [ ] Integrate with [Colossal-AI](https://github.com/PKU-YuanGroup/Open-Sora-Plan/issues/59#issue-2170735221) for a cheaper, faster, and more efficient. ๐Ÿ™ **[Need your contribution]** - [ ] Train with a text condition. Here we could conduct different experiments: - [ ] Train with T5 conditioning. ๐Ÿš€ **[Require more computation]**