ColossalAI/colossalai/booster
YeAnbang ed43a4be04
[Distributed RLHF] Integration of PP (#6257)
* update help information

* update style

* fix

* minor fix

* support PP training

* add pp support

* remove unused code

* address conversation

---------

Co-authored-by: Tong Li <tong.li35271158@gmail.com>
2025-04-09 13:23:24 +08:00
..
mixed_precision [Zerobubble] merge main. (#6142) 2024-11-19 19:00:36 +08:00
plugin [Distributed RLHF] Integration of PP (#6257) 2025-04-09 13:23:24 +08:00
__init__.py [booster] implemented the torch ddd + resnet example (#3232) 2023-03-27 10:24:14 +08:00
accelerator.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
booster.py [checkpointio] support non blocking pin load (#6172) 2024-12-25 17:03:25 +08:00