github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-16 17:16:14 +00:00

Files

History

Tong Li abca66e69f fix reward score

2025-03-11 10:17:32 +08:00

..

fix reward score

2025-03-11 10:17:32 +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

[pre-commit.ci] auto fixes from pre-commit.com hooks

2025-03-06 08:29:59 +00:00

grpo_consumer.py

update grpo

2025-03-10 14:12:04 +08:00

inference_backend.py

polish

2025-02-28 10:16:42 +08:00

launch.py

add algo selection

2025-03-06 14:29:22 +08:00

loss.py

update loss

2025-03-06 10:53:03 +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