Skip to content

Commit 77e0969

Browse files
committed
fix
1 parent a7315bc commit 77e0969

File tree

3 files changed

+3
-2
lines changed

3 files changed

+3
-2
lines changed

.github/workflows/unittest-gpu.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -135,6 +135,7 @@ jobs:
135135
docker exec -t $container_name /bin/bash -c '
136136
cd /workspace/PaddleNLP
137137
rm -rf tests/utils/test_aistudio_download.py
138+
'
138139
fi
139140
140141
- name: Test

tests/llm/test_grpo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def test_grpo(self):
9292
# 运行主逻辑
9393
cmd = 'python -u -m paddle.distributed.launch \
9494
--devices "$CUDA_VISIBLE_DEVICES" run_rl.py \
95-
../../config/qwen/reinforce_plus_plus_argument.yaml \
95+
../../config/qwen/grpo_argument.yaml \
9696
--actor_model_name_or_path "Qwen/Qwen2-1.5B" \
9797
--max_dec_len 128 \
9898
--max_steps 3 \

tests/llm/test_reinforce_plus_plus.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def test_reinforce_plus_plus(self):
9292
# 运行主逻辑
9393
cmd = 'python -u -m paddle.distributed.launch \
9494
--devices "$CUDA_VISIBLE_DEVICES" run_rl.py \
95-
../../config/qwen/reinforce_plus_plus_argument.yaml \
95+
../../config/qwen/grpo_argument.yaml \
9696
--rl_algorithm "reinforce_plus_plus" \
9797
--actor_model_name_or_path "Qwen/Qwen2-1.5B" \
9898
--max_dec_len 128 \

0 commit comments

Comments
 (0)