mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-17 12:59:50 +00:00

Files

Tong Li 32afa7bf29 fix empty tensor (#6319 )

Co-authored-by: Tong Li <tong.li35271158@gmail.com>

2025-05-20 17:41:44 +08:00

upgrade reward functions

2025-05-16 18:04:38 +08:00

__init__.py

2025-02-21 15:24:23 +08:00

comm.py

2025-02-21 15:24:23 +08:00

consumer.py

handle empty index

2025-05-15 20:32:05 +08:00

grpo_consumer.py

fix empty tensor (#6319 )

2025-05-20 17:41:44 +08:00

inference_backend.py

move logging to producer

2025-05-14 18:10:57 +08:00

launch.py

add uuid to rollout log

2025-05-20 09:45:56 +08:00

loss.py

2025-04-25 17:39:17 +08:00

producer.py

fix logging rollouts

2025-05-17 21:12:58 +08:00

README.md

2025-02-21 15:24:23 +08:00

utils.py

move logging to producer

2025-05-14 18:10:57 +08:00

Requirements

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