github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-15 06:54:52 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
de0c267f5a6e67103a3fc643fb0bc5798ed21ea1
ColossalAI/applications/ColossalChat/coati/distributed
History
YeAnbang de0c267f5a reuse comm-group
2025-08-05 13:59:02 +08:00
..
reward
Support evaluation during training
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
reuse comm-group
2025-08-05 13:59:02 +08:00
grpo_consumer.py
Support evaluation during training
2025-08-05 13:59:02 +08:00
inference_backend.py
Support evaluation during training
2025-08-05 13:59:02 +08:00
launch.py
Support evaluation during training
2025-08-05 13:59:02 +08:00
loss.py
[feat] Support DAPO (#6263)
2025-08-05 13:59:02 +08:00
producer.py
reuse comm-group
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
Support evaluation during training
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
Reference in New Issue View Git Blame Copy Permalink