mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-12 19:06:17 +00:00

Files

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

for more information, see https://pre-commit.ci

2025-08-05 14:04:08 +08:00

upgrade reward functions

2025-08-05 14:01:20 +08:00

__init__.py

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

comm.py

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

consumer.py

2025-08-05 14:04:08 +08:00

grpo_consumer.py

2025-08-05 14:04:08 +08:00

inference_backend.py

move logging to producer

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

launch.py

2025-08-05 14:01:45 +08:00

loss.py

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

producer.py

address conversation

2025-08-05 14:02:36 +08:00

README.md

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

utils.py

move logging to producer

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

Requirements

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