tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

GRPO (#2307)

Browse Source
This commit is contained in:
Wing Lian
2025-02-13 16:01:01 -05:00
committed by GitHub
parent fdbb1a207c
commit ffae8d6a95
28 changed files with 900 additions and 183 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
.github/workflows/multi-gpu-e2e.yml vendored
View File

@@ -24,20 +24,21 @@ jobs:
cuda_version: 12.4.1
python_version: "3.11"
pytorch: 2.4.1
axolotl_extras:
axolotl_extras: # no vllm support for 2.4.1
num_gpus: 2
nightly_build: "true"
- cuda: 124
cuda_version: 12.4.1
python_version: "3.11"
pytorch: 2.5.1
axolotl_extras:
axolotl_extras: vllm
num_gpus: 2
nightly_build: "true"
- cuda: 124
cuda_version: 12.4.1
python_version: "3.11"
pytorch: 2.6.0
# awaiting vllm#12721
axolotl_extras:
num_gpus: 2
nightly_build: "true"