Merge branch 'grpo-latest-rebase-main' of https://github.com/hpcaitech/ColossalAI into grpo-latest-rebase-main

This commit is contained in:
YeAnbang
2025-08-14 19:03:04 +08:00
4 changed files with 12 additions and 4 deletions

View File

@@ -21,7 +21,7 @@ jobs:
container:
image: image-cloud.luchentech.com/hpcaitech/pytorch-cuda:2.2.2-12.1.0
options: --gpus all --rm -v /data/scratch/examples-data:/data/scratch/examples-data --shm-size=10.24gb
timeout-minutes: 60
timeout-minutes: 180
defaults:
run:
shell: bash
@@ -34,6 +34,10 @@ jobs:
pip install --no-cache-dir -v -e .
- name: Install ChatGPT
env:
CFLAGS: "-O1"
CXXFLAGS: "-O1"
MAX_JOBS: 4
run: |
pip install flash-attn --no-build-isolation
cd applications/ColossalChat