github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-16 17:16:14 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
070907dd7fe9d2dcf32e06b928a54a73b406bd83
ColossalAI/applications/ColossalChat/coati/distributed
History
Tong Li 070907dd7f polish
2025-02-28 10:16:42 +08:00
..
reward
polish
2025-02-28 10:16:42 +08:00
__init__.py
[chat] add distributed impl (#6210)
2025-02-21 15:24:23 +08:00
comm.py
[chat] add distributed impl (#6210)
2025-02-21 15:24:23 +08:00
consumer.py
[feature] fit RL style generation (#6213)
2025-02-21 17:28:19 +08:00
grpo_consumer.py
polish
2025-02-28 10:16:42 +08:00
inference_backend.py
polish
2025-02-28 10:16:42 +08:00
launch.py
add simple grpo
2025-02-23 22:54:26 +08:00
loss.py
add simple grpo
2025-02-23 22:54:26 +08:00
producer.py
[chat] add distributed impl (#6210)
2025-02-21 15:24:23 +08:00
README.md
[chat] add distributed impl (#6210)
2025-02-21 15:24:23 +08:00
utils.py
add simple grpo
2025-02-23 22:54:26 +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