Skip to content

Commit 3340b3a

Browse files
committed
Change model to Qwen/Qwen2.5-0.5B-Instruct for CI.
Signed-off-by: Jonas Yang <joyang@nvidia.com>
1 parent a123e2d commit 3340b3a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.github/workflows/e2e_ppo_grpo_trainer_trtllm.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ jobs:
137137
data.train_files="['${HOME}/data/gsm8k/train.parquet']" \
138138
data.val_files="['${HOME}/data/gsm8k/test.parquet']" \
139139
trainer.logger='["console"]' \
140-
actor_rollout_ref.model.path=Qwen/Qwen2-0.5B-Instruct
140+
actor_rollout_ref.model.path=Qwen/Qwen2.5-0.5B-Instruct
141141
- name: clean up
142142
run: |
143143
rm -rf checkpoints
@@ -174,7 +174,7 @@ jobs:
174174
data.train_files="['${HOME}/data/gsm8k/train.parquet']" \
175175
data.val_files="['${HOME}/data/gsm8k/test.parquet']" \
176176
trainer.logger='["console"]' \
177-
actor_rollout_ref.model.path=Qwen/Qwen2-0.5B-Instruct
177+
actor_rollout_ref.model.path=Qwen/Qwen2.5-0.5B-Instruct
178178
- name: clean up
179179
run: |
180180
rm -rf checkpoints

0 commit comments

Comments
 (0)