Files
ColossalAI/applications/ColossalChat/rl_example.py
YeAnbang 3bd6fa3c67 [hot-fix] Fix memory leakage bug, support TP+PP (#6258)
* update help information

* update style

* fix

* minor fix

* support PP training

* add pp support

* remove unused code

* address conversation

* fix memory leakage support tp+pp

* move empty cache

* move empty cache

---------

Co-authored-by: Tong Li <tong.li35271158@gmail.com>
2025-08-05 13:59:02 +08:00

5.1 KiB