github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-26 17:53:08 +00:00

Files

History

YeAnbang 0d0fef771f disable wandb tb syncing

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

..

Support evaluation during training

2025-08-05 13:59:03 +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

[feat] Support prompt level dynamic (#6300 )

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

grpo_consumer.py

disable wandb tb syncing

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

inference_backend.py

move logging to producer

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

launch.py

use consumer global step

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

loss.py

[feat] Support DAPO (#6263 )

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

producer.py

disable wandb tb syncing

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

README.md

[chat] add distributed impl (#6210 )

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

utils.py

move logging to producer

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

README.md

Requirements

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