Skip to content

Commit 8c85eb0

Browse files
authored
Merge branch 'volcengine:main' into main
2 parents 7be1823 + d640f99 commit 8c85eb0

201 files changed

Lines changed: 7269 additions & 2615 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

.github/PULL_REQUEST_TEMPLATE.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,4 +37,4 @@
3737
- [ ] Apply [pre-commit checks](https://github.com/volcengine/verl/blob/main/CONTRIBUTING.md#code-linting-and-formatting): `pre-commit install && pre-commit run --all-files --show-diff-on-failure --color=always`
3838
- [ ] Add / Update [the documentation](https://github.com/volcengine/verl/tree/main/docs).
3939
- [ ] Add unit or end-to-end test(s) to [the CI workflow](https://github.com/volcengine/verl/tree/main/.github/workflows) to cover all the code. If not feasible, explain why: ...
40-
- [ ] Once your PR is ready for CI, send a message in [the `ci-request` channel](https://verl-project.slack.com/archives/C091TCESWB1) in [the `verl` Slack workspace](https://join.slack.com/t/verl-project/shared_invite/zt-3855yhg8g-CTkqXu~hKojPCmo7k_yXTQ).
40+
- [ ] Once your PR is ready for CI, send a message in [the `ci-request` channel](https://verl-project.slack.com/archives/C091TCESWB1) in [the `verl` Slack workspace](https://join.slack.com/t/verl-project/shared_invite/zt-3855yhg8g-CTkqXu~hKojPCmo7k_yXTQ). (If not accessible, please try [the Feishu group (飞书群)](https://applink.larkoffice.com/client/chat/chatter/add_by_link?link_token=772jd4f1-cd91-441e-a820-498c6614126a).)

.github/workflows/checkpoint_converter.yml

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
# - `special_sanity`: a suite of quick sanity tests
1313
# - `special_standalone`: a set of test that are designed to run in dedicated environments
1414

15-
# Accelerators for tests
15+
# Accelerators for tests
1616
# - By default tests are run with GPU available, except for the ones under `special_npu`, and any test script whose name ends with `on_cpu.py`.
1717
# - For test scripts with `on_cpu.py` name suffix would be tested on CPU resources in linux environment.
1818

@@ -29,8 +29,6 @@
2929
# - new workflow yaml is added to `.github/workflows`
3030
# - new tests are added to workflow mentioned in 2.
3131

32-
33-
3432
name: checkpoint_converter
3533
# latest version: Megatron-LM core_r0.11.0 https://github.com/NVIDIA/Megatron-LM/tree/core_r0.11.0
3634

@@ -64,14 +62,13 @@ on:
6462
- "verl/trainer/main_ppo.py"
6563
- "verl/trainer/config/ppo_megatron_trainer.yaml"
6664

67-
6865
# Cancel jobs on the same ref if a new one is triggered
6966
concurrency:
7067
group: ${{ github.workflow }}-${{ github.ref }}
7168
cancel-in-progress: ${{ github.ref != 'refs/heads/main' }}
7269

7370
# Declare permissions just read content.
74-
permissions:
71+
permissions:
7572
contents: read
7673

7774
jobs:
@@ -84,15 +81,20 @@ jobs:
8481
NO_PROXY: "localhost,127.0.0.1"
8582
HF_HUB_ENABLE_HF_TRANSFER: "0" # This is more stable
8683
container:
87-
image: verlai/verl:app-verl0.4-sglang0.4.6.post5-vllm0.8.5-mcore0.12.2-te2.2
84+
image: verlai/verl:app-verl0.5-sglang0.4.8-mcore0.12.2-te2.2
8885
options: --gpus all --shm-size=10g
8986
steps:
9087
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
9188
with:
92-
fetch-depth: 0
89+
fetch-depth: 0
9390
- name: Install the current repository
9491
run: |
9592
pip3 install -e .[test]
93+
- name: Download Model to Use
94+
run: |
95+
huggingface-cli download Qwen/Qwen2.5-0.5B --local-dir ${HOME}/models/Qwen/Qwen2.5-0.5B
96+
huggingface-cli download deepseek-ai/deepseek-coder-1.3b-instruct --local-dir ${HOME}/models/deepseek-ai/deepseek-coder-1.3b-instruct
97+
export HF_HUB_OFFLINE=1
9698
- name: Running Huggingface to Megatron dist_ckpt converter (Qwen/Qwen2.5-0.5B)
9799
run: |
98100
ray stop --force
@@ -114,12 +116,12 @@ jobs:
114116
HF_HUB_ENABLE_HF_TRANSFER: "0" # This is more stable
115117
HF_ENDPOINT: "https://hf-mirror.com"
116118
container:
117-
image: verlai/verl:app-verl0.4-sglang0.4.6.post5-vllm0.8.5-mcore0.12.2-te2.2
119+
image: verlai/verl:app-verl0.5-sglang0.4.8-mcore0.12.2-te2.2
118120
options: --gpus all --shm-size=10g
119121
steps:
120122
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
121123
with:
122-
fetch-depth: 0
124+
fetch-depth: 0
123125
- name: Install the current repository
124126
run: |
125127
pip3 install -e .[test]

.github/workflows/cpu_unit_tests.yml

Lines changed: 12 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -59,17 +59,20 @@ permissions:
5959

6060
jobs:
6161
cpu_unit_tests:
62-
runs-on: ubuntu-latest
63-
timeout-minutes: 10 # Increase this timeout value as needed
64-
strategy:
65-
matrix:
66-
python-version: ["3.10"]
62+
runs-on: [L20x8]
63+
timeout-minutes: 20 # Increase this timeout value as needed
64+
env:
65+
HTTP_PROXY: ${{ secrets.PROXY_HTTP }}
66+
HTTPS_PROXY: ${{ secrets.PROXY_HTTPS }}
67+
NO_PROXY: "localhost,127.0.0.1,hf-mirror.com"
68+
HF_ENDPOINT: "https://hf-mirror.com"
69+
HF_HUB_ENABLE_HF_TRANSFER: "0" # This is more stable
70+
container:
71+
image: verlai/verl:app-verl0.5-vllm0.9.1-mcore0.12.2-te2.2
6772
steps:
6873
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
69-
- name: Set up Python ${{ matrix.python-version }}
70-
uses: actions/setup-python@0b93645e9fea7318ecaed2b359559ac225c90a2b # v5.3.0
7174
with:
72-
python-version: ${{ matrix.python-version }}
75+
fetch-depth: 0
7376
- name: Install the current repository
7477
run: |
7578
pip install -e .[test,prime,geo]
@@ -80,4 +83,4 @@ jobs:
8083
python3 examples/data_preprocess/geo3k.py
8184
echo '[pytest]' > pytest.ini
8285
echo 'python_files = *_on_cpu.py' >> pytest.ini
83-
pytest -s -x --asyncio-mode=auto tests/
86+
pytest -s -x --asyncio-mode=auto tests/

.github/workflows/e2e_ascend.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ jobs:
7272
test:
7373
name: verl Ascend test (self-host)
7474
runs-on: [self-hosted, npu-0]
75-
timeout-minutes: 30 # Increase this timeout value as needed
75+
timeout-minutes: 40 # Increase this timeout value as needed
7676
container:
7777
image: crispig/verl_npu:cann8.1rc1-py3.10-torch2.5.1-vllm-ascend0.7.3.post1-250616
7878
volumes:

.github/workflows/e2e_dapo.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@ jobs:
9494
HF_ENDPOINT: "https://hf-mirror.com"
9595
HF_HUB_ENABLE_HF_TRANSFER: "0" # This is more stable
9696
container:
97-
image: verlai/verl:app-verl0.4-vllm0.8.5-mcore0.12.2-te2.2
97+
image: verlai/verl:app-verl0.5-vllm0.9.1-mcore0.12.2-te2.2
9898
options: --gpus all --shm-size=10g
9999
steps:
100100
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2

.github/workflows/e2e_eval_aime24.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ permissions:
8888
contents: read
8989

9090
env:
91-
IMAGE: "verl-ci-cn-beijing.cr.volces.com/verlai/verl:app-verl0.4-vllm0.8.5-mcore0.12.2-te2.2"
91+
IMAGE: "verl-ci-cn-beijing.cr.volces.com/verlai/verl:app-verl0.5-vllm0.9.1-mcore0.12.2-te2.2"
9292
DYNAMIC_RUNNER_ENDPOINT: "https://sd10g3clalm04ug7alq90.apigateway-cn-beijing.volceapi.com/runner"
9393

9494
jobs:

.github/workflows/e2e_ppo_trainer.yml

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ jobs:
8787
HF_ENDPOINT: "https://hf-mirror.com"
8888
HF_HUB_ENABLE_HF_TRANSFER: "0" # This is more stable
8989
container:
90-
image: verlai/verl:app-verl0.4-vllm0.8.5-mcore0.12.2-te2.2
90+
image: verlai/verl:app-verl0.5-vllm0.9.1-mcore0.12.2-te2.2
9191
options: --gpus all --shm-size=10g
9292
steps:
9393
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
@@ -211,6 +211,12 @@ jobs:
211211
run: |
212212
ray stop --force
213213
FUSED_KERNEL=True FUSED_KERNEL_BACKEND=triton bash tests/special_e2e/ppo_trainer/run_model_reward.sh
214+
- name: Running GSM8K E2E training tests on vllm async
215+
run: |
216+
ray stop --force
217+
export VLLM_USE_V1=1
218+
ray start --head
219+
TOTAL_TRAIN_STEPS=2 ENGINE=vllm ROLLOUT_MODE=async bash tests/special_e2e/ppo_trainer/run_function_reward.sh
214220
215221
e2e_ppo_trainer_vllm_vlm:
216222
runs-on: [L20x8]
@@ -223,15 +229,15 @@ jobs:
223229
HF_ENDPOINT: "https://hf-mirror.com"
224230
HF_HUB_ENABLE_HF_TRANSFER: "0" # This is more stable
225231
container:
226-
image: verlai/verl:app-verl0.4-vllm0.8.5-mcore0.12.2-te2.2
232+
image: verlai/verl:app-verl0.5-vllm0.9.1-mcore0.12.2-te2.2
227233
options: --gpus all --shm-size=50g # Visual dataloader requires large memory
228234
steps:
229235
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
230236
with:
231237
fetch-depth: 0
232238
- name: Install the current repository
233239
run: |
234-
pip3 install -e .[test,gpu,vllm,geo,trl]
240+
pip3 install --no-deps -e .[test,gpu,vllm,geo,trl]
235241
pip install "transformers[hf_xet]<4.53.0" # Fix for transformers 4.53.0
236242
# Geo3k
237243
- name: Prepare GEO3K dataset
@@ -280,7 +286,7 @@ jobs:
280286
HF_ENDPOINT: "https://hf-mirror.com"
281287
HF_HUB_ENABLE_HF_TRANSFER: "0" # This is more stable
282288
container:
283-
image: verlai/verl:app-verl0.4-sglang0.4.6.post5-vllm0.8.5-mcore0.12.2-te2.2
289+
image: verlai/verl:app-verl0.5-sglang0.4.8-mcore0.12.2-te2.2
284290
options: --gpus all --shm-size=10g
285291
steps:
286292
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
@@ -301,12 +307,6 @@ jobs:
301307
run: |
302308
ray stop --force
303309
TOTAL_TRAIN_STEPS=2 ENGINE=sglang ROLLOUT_MODE=async bash tests/special_e2e/ppo_trainer/run_function_reward.sh
304-
- name: Running GSM8K E2E training tests on vllm async
305-
run: |
306-
ray stop --force
307-
export VLLM_USE_V1=1
308-
ray start --head
309-
TOTAL_TRAIN_STEPS=2 ENGINE=vllm ROLLOUT_MODE=async bash tests/special_e2e/ppo_trainer/run_function_reward.sh
310310
311311
e2e_ppo_trainer_sglang_multiturn_with_tool:
312312
runs-on: [L20x8]
@@ -319,7 +319,7 @@ jobs:
319319
HF_ENDPOINT: "https://hf-mirror.com"
320320
HF_HUB_ENABLE_HF_TRANSFER: "0" # This is more stable
321321
container:
322-
image: verlai/verl:app-verl0.4-sglang0.4.6.post5-vllm0.8.5-mcore0.12.2-te2.2
322+
image: verlai/verl:app-verl0.5-sglang0.4.8-mcore0.12.2-te2.2
323323
options: --gpus all --shm-size=10g
324324
steps:
325325
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
@@ -352,7 +352,7 @@ jobs:
352352
HF_ENDPOINT: "https://hf-mirror.com"
353353
HF_HUB_ENABLE_HF_TRANSFER: "0" # This is more stable
354354
container:
355-
image: verlai/verl:app-verl0.4-sglang0.4.6.post5-vllm0.8.5-mcore0.12.2-te2.2
355+
image: verlai/verl:app-verl0.5-sglang0.4.8-mcore0.12.2-te2.2
356356
options: --gpus all --shm-size=50g # Visual dataloader requires large memory
357357
steps:
358358
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
@@ -409,7 +409,7 @@ jobs:
409409
HF_ENDPOINT: "https://hf-mirror.com"
410410
HF_HUB_ENABLE_HF_TRANSFER: "0" # This is more stable
411411
container:
412-
image: verlai/verl:app-verl0.4-sglang0.4.6.post5-vllm0.8.5-mcore0.12.2-te2.2
412+
image: verlai/verl:app-verl0.5-sglang0.4.8-mcore0.12.2-te2.2
413413
options: --gpus all --shm-size=10g
414414
steps:
415415
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2

.github/workflows/e2e_ppo_trainer_megatron_sglang.yml

Lines changed: 20 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
# - `special_sanity`: a suite of quick sanity tests
1313
# - `special_standalone`: a set of test that are designed to run in dedicated environments
1414

15-
# Accelerators for tests
15+
# Accelerators for tests
1616
# - By default tests are run with GPU available, except for the ones under `special_npu`, and any test script whose name ends with `on_cpu.py`.
1717
# - For test scripts with `on_cpu.py` name suffix would be tested on CPU resources in linux environment.
1818

@@ -85,7 +85,7 @@ permissions:
8585
contents: read
8686

8787
env:
88-
IMAGE: "verl-ci-cn-beijing.cr.volces.com/verlai/verl:app-verl0.4-sglang0.4.6.post5-vllm0.8.5-mcore0.12.2-te2.2"
88+
IMAGE: "verl-ci-cn-beijing.cr.volces.com/verlai/verl:app-verl0.5-sglang0.4.8-mcore0.12.2-te2.2"
8989
DYNAMIC_RUNNER_ENDPOINT: "https://sd10g3clalm04ug7alq90.apigateway-cn-beijing.volceapi.com/runner"
9090

9191
jobs:
@@ -98,7 +98,7 @@ jobs:
9898
steps:
9999
- uses: actions/checkout@v4
100100
- id: create-runner
101-
uses: volcengine/vemlp-github-runner@v1
101+
uses: volcengine/vemlp-github-runner@v1
102102
with:
103103
mode: "create"
104104
faas-url: "${{ env.DYNAMIC_RUNNER_ENDPOINT }}"
@@ -234,6 +234,10 @@ jobs:
234234
- name: Prepare GSM8K dataset
235235
run: |
236236
python3 examples/data_preprocess/gsm8k.py
237+
- name: Download Model to Use
238+
run: |
239+
huggingface-cli download Qwen/Qwen2.5-0.5B --local-dir ${HOME}/models/Qwen/Qwen2.5-0.5B
240+
export HF_HUB_OFFLINE=1
237241
- name: Prepare dist_ckpt of Qwen2.5-0.5B, uneven layer distribution only supports dist_ckpt
238242
run: |
239243
python3 scripts/converter_hf_to_mcore.py --hf_model_path ${HOME}/models/Qwen/Qwen2.5-0.5B --output_path checkpoints/verl-test/qwen2.5-0.5b-megatron
@@ -341,26 +345,29 @@ jobs:
341345
- name: Running Geo3k E2E training tests with 3D parallelism on 8 L20 GPUs with Megatron (Qwen)
342346
run: |
343347
ray stop --force
344-
ENGINE=sglang TRAIN_FILES=${HOME}/data/geo3k/train.parquet VAL_FILES=${HOME}/data/geo3k/test.parquet MAX_PROMPT_LENGTH=1024 MAX_RESPONSE_LENGTH=2048 MODEL_ID=Qwen/Qwen2.5-VL-3B-Instruct ADV_ESTIMATOR=grpo USE_DYNAMIC_BSZ=False SKIP_SAVE_HF_MODEL=1 COMMON_PP=4 COMMON_VPP=null COMMON_CP=1 COMMON_TP=2 USE_DIST_CKPT=true DIST_CKPT_PATH=checkpoints/verl-test/qwen2.5-vl-3b-megatron bash tests/special_e2e/run_ppo_trainer_megatron.sh
348+
ENGINE=sglang TRAIN_FILES=${HOME}/data/geo3k/train.parquet VAL_FILES=${HOME}/data/geo3k/test.parquet MAX_PROMPT_LENGTH=1024 MAX_RESPONSE_LENGTH=2048 MODEL_ID=Qwen/Qwen2.5-VL-3B-Instruct ADV_ESTIMATOR=grpo USE_DYNAMIC_BSZ=False SKIP_SAVE_HF_MODEL=1 COMMON_PP=4 COMMON_VPP=null COMMON_CP=1 COMMON_TP=2 USE_DIST_CKPT=true DIST_CKPT_PATH=checkpoints/verl-test/qwen2.5-vl-3b-megatron bash tests/special_e2e/run_ppo_trainer_megatron.sh
345349
- name: clean up
346350
run: |
347351
rm -rf checkpoints
348352
349353
cleanup:
350354
runs-on: ubuntu-latest
351-
needs: [setup,
352-
e2e_ppo_trainer_megatron-deepseek,
353-
e2e_ppo_trainer_megatron-qwen3,
354-
e2e_ppo_trainer_megatron-different-train-infer-tp-qwen-tie-embedding,
355-
e2e_ppo_trainer_megatron-qwen-override-transformer-config,
356-
e2e_ppo_trainer_megatron-deepseek-override-transformer-config,
357-
e2e_ppo_trainer_megatron-moe-expert-parallel,
358-
e2e_ppo_trainer_megatron-qwen2_5vl-3b]
355+
needs:
356+
[
357+
setup,
358+
e2e_ppo_trainer_megatron-deepseek,
359+
e2e_ppo_trainer_megatron-qwen3,
360+
e2e_ppo_trainer_megatron-different-train-infer-tp-qwen-tie-embedding,
361+
e2e_ppo_trainer_megatron-qwen-override-transformer-config,
362+
e2e_ppo_trainer_megatron-deepseek-override-transformer-config,
363+
e2e_ppo_trainer_megatron-moe-expert-parallel,
364+
e2e_ppo_trainer_megatron-qwen2_5vl-3b,
365+
]
359366
if: always()
360367
steps:
361368
- id: destroy-runner
362369
uses: volcengine/vemlp-github-runner@v1
363370
with:
364371
mode: "destroy"
365372
faas-url: "${{ env.DYNAMIC_RUNNER_ENDPOINT }}"
366-
mlp-task-id: "${{ needs.setup.outputs.mlp-task-id }}"
373+
mlp-task-id: "${{ needs.setup.outputs.mlp-task-id }}"

.github/workflows/e2e_ppo_trainer_megatron_vllm.yml

Lines changed: 21 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
# - `special_sanity`: a suite of quick sanity tests
1313
# - `special_standalone`: a set of test that are designed to run in dedicated environments
1414

15-
# Accelerators for tests
15+
# Accelerators for tests
1616
# - By default tests are run with GPU available, except for the ones under `special_npu`, and any test script whose name ends with `on_cpu.py`.
1717
# - For test scripts with `on_cpu.py` name suffix would be tested on CPU resources in linux environment.
1818

@@ -85,7 +85,7 @@ permissions:
8585
contents: read
8686

8787
env:
88-
IMAGE: "verl-ci-cn-beijing.cr.volces.com/verlai/verl:app-verl0.4-vllm0.8.5-mcore0.12.2-te2.2"
88+
IMAGE: "verl-ci-cn-beijing.cr.volces.com/verlai/verl:app-verl0.5-vllm0.9.1-mcore0.12.2-te2.2"
8989
DYNAMIC_RUNNER_ENDPOINT: "https://sd10g3clalm04ug7alq90.apigateway-cn-beijing.volceapi.com/runner"
9090

9191
jobs:
@@ -98,7 +98,7 @@ jobs:
9898
steps:
9999
- uses: actions/checkout@v4
100100
- id: create-runner
101-
uses: volcengine/vemlp-github-runner@v1
101+
uses: volcengine/vemlp-github-runner@v1
102102
with:
103103
mode: "create"
104104
faas-url: "${{ env.DYNAMIC_RUNNER_ENDPOINT }}"
@@ -214,7 +214,7 @@ jobs:
214214
- name: Running GSM8K E2E training tests with 3D parallelism on 8 L20 GPUs with Megatron (Qwen) with train tp < infer tp
215215
run: |
216216
ray stop --force
217-
VAL_BEFORE_TRAIN=True TEST_FREQ=1 SAVE_FREQ=1 TRAIN_TP=1 INFER_TP=2 MODEL_ID=Qwen/Qwen2.5-1.5B bash tests/special_e2e/run_ppo_trainer_megatron.sh
217+
VAL_BEFORE_TRAIN=True TEST_FREQ=1 SAVE_FREQ=1 TRAIN_TP=1 INFER_TP=2 ALL_OFFLOAD=True MODEL_ID=Qwen/Qwen2.5-1.5B bash tests/special_e2e/run_ppo_trainer_megatron.sh
218218
- name: clean up
219219
run: |
220220
rm -rf checkpoints
@@ -238,6 +238,10 @@ jobs:
238238
- name: Prepare GSM8K dataset
239239
run: |
240240
python3 examples/data_preprocess/gsm8k.py
241+
- name: Download Model to Use
242+
run: |
243+
huggingface-cli download Qwen/Qwen2.5-0.5B --local-dir ${HOME}/models/Qwen/Qwen2.5-0.5B
244+
export HF_HUB_OFFLINE=1
241245
- name: Prepare dist_ckpt of Qwen2.5-0.5B, uneven layer distribution only supports dist_ckpt
242246
run: |
243247
python3 scripts/converter_hf_to_mcore.py --hf_model_path ${HOME}/models/Qwen/Qwen2.5-0.5B --output_path checkpoints/verl-test/qwen2.5-0.5b-megatron
@@ -347,26 +351,29 @@ jobs:
347351
- name: Running Geo3k E2E training tests with 3D parallelism on 8 L20 GPUs with Megatron (Qwen)
348352
run: |
349353
ray stop --force
350-
TRAIN_FILES=${HOME}/data/geo3k/train.parquet VAL_FILES=${HOME}/data/geo3k/test.parquet MAX_PROMPT_LENGTH=1024 MAX_RESPONSE_LENGTH=2048 MODEL_ID=Qwen/Qwen2.5-VL-3B-Instruct ADV_ESTIMATOR=grpo USE_DYNAMIC_BSZ=False SKIP_SAVE_HF_MODEL=1 COMMON_PP=4 COMMON_VPP=null COMMON_CP=1 COMMON_TP=2 USE_DIST_CKPT=true DIST_CKPT_PATH=checkpoints/verl-test/qwen2.5-vl-3b-megatron bash tests/special_e2e/run_ppo_trainer_megatron.sh
354+
TRAIN_FILES=${HOME}/data/geo3k/train.parquet VAL_FILES=${HOME}/data/geo3k/test.parquet MAX_PROMPT_LENGTH=1024 MAX_RESPONSE_LENGTH=2048 MODEL_ID=Qwen/Qwen2.5-VL-3B-Instruct ADV_ESTIMATOR=grpo USE_DYNAMIC_BSZ=False SKIP_SAVE_HF_MODEL=1 COMMON_PP=4 COMMON_VPP=null COMMON_CP=1 COMMON_TP=2 USE_DIST_CKPT=true DIST_CKPT_PATH=checkpoints/verl-test/qwen2.5-vl-3b-megatron bash tests/special_e2e/run_ppo_trainer_megatron.sh
351355
- name: clean up
352356
run: |
353357
rm -rf checkpoints
354358
355359
cleanup:
356360
runs-on: ubuntu-latest
357-
needs: [setup,
358-
e2e_ppo_trainer_megatron-deepseek,
359-
e2e_ppo_trainer_megatron-qwen3,
360-
e2e_ppo_trainer_megatron-different-train-infer-tp-qwen-tie-embedding,
361-
e2e_ppo_trainer_megatron-qwen-override-transformer-config,
362-
e2e_ppo_trainer_megatron-deepseek-override-transformer-config,
363-
e2e_ppo_trainer_megatron-moe-expert-parallel,
364-
e2e_ppo_trainer_megatron-qwen2_5vl-3b]
361+
needs:
362+
[
363+
setup,
364+
e2e_ppo_trainer_megatron-deepseek,
365+
e2e_ppo_trainer_megatron-qwen3,
366+
e2e_ppo_trainer_megatron-different-train-infer-tp-qwen-tie-embedding,
367+
e2e_ppo_trainer_megatron-qwen-override-transformer-config,
368+
e2e_ppo_trainer_megatron-deepseek-override-transformer-config,
369+
e2e_ppo_trainer_megatron-moe-expert-parallel,
370+
e2e_ppo_trainer_megatron-qwen2_5vl-3b,
371+
]
365372
if: always()
366373
steps:
367374
- id: destroy-runner
368375
uses: volcengine/vemlp-github-runner@v1
369376
with:
370377
mode: "destroy"
371378
faas-url: "${{ env.DYNAMIC_RUNNER_ENDPOINT }}"
372-
mlp-task-id: "${{ needs.setup.outputs.mlp-task-id }}"
379+
mlp-task-id: "${{ needs.setup.outputs.mlp-task-id }}"

0 commit comments

Comments
 (0)