mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-14 13:48:15 +00:00

Files

Tong Li 6a1bd833e0 [feat] Sync shard model (#6289 )

* [feat] support hybrid parallel model sync

* update consumer and producer

* update files

* update producer

* remove print

* update

---------

Co-authored-by: duanjunwen <935724073@qq.com>
Co-authored-by: YeAnbang <44796419+YeAnbang@users.noreply.github.com>
Co-authored-by: Tong Li <tong.li35271158@gmail.com>

2025-08-05 13:59:02 +08:00

reward

[feat] Support boxed math reward (#6284 )

2025-08-05 13:59:02 +08:00

__init__.py

[chat] add distributed impl (#6210 )

2025-08-05 13:59:01 +08:00

comm.py

[chat] add distributed impl (#6210 )

2025-08-05 13:59:01 +08:00

consumer.py

[feat] Sync shard model (#6289 )

2025-08-05 13:59:02 +08:00

grpo_consumer.py

[feat] Support boxed math reward (#6284 )

2025-08-05 13:59:02 +08:00

inference_backend.py

[feat] Support DAPO (#6263 )

2025-08-05 13:59:02 +08:00

launch.py

[feat] Sync shard model (#6289 )

2025-08-05 13:59:02 +08:00

loss.py

[feat] Support DAPO (#6263 )

2025-08-05 13:59:02 +08:00

producer.py

[feat] Sync shard model (#6289 )

2025-08-05 13:59:02 +08:00

README.md

[chat] add distributed impl (#6210 )

2025-08-05 13:59:01 +08:00

utils.py

[feat] Support DAPO (#6263 )

2025-08-05 13:59:02 +08:00

README.md

Requirements

pip install cupy-cuda12x
python -m cupyx.tools.install_library --cuda 12.x --library nccl