Skip to content

Commit 0b57642

Browse files
author
Vincent Moens
committed
[CI] Fix LLM tests
ghstack-source-id: 0ef6d2c Pull Request resolved: #2918
1 parent b247526 commit 0b57642

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

.github/unittest/linux_libs/scripts_llm/run_test.sh

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -27,13 +27,10 @@ python -c "import transformers, datasets"
2727

2828
pytest test/test_rlhf.py --instafail -v --durations 200 --capture no --error-for-skips
2929

30-
pytest examples/rlhf/train_rlhf.py \
30+
python examples/rlhf/train_rlhf.py \
3131
sys.device=cuda:0 sys.ref_device=cuda:0 \
3232
model.name_or_path=gpt2 train.max_epochs=2 \
3333
data.batch_size=2 train.ppo.ppo_batch_size=2 \
3434
train.ppo.ppo_num_epochs=1 reward_model.name_or_path= \
3535
train.ppo.episode_length=8 train.ppo.num_rollouts_per_epoch=4 \
3636
data.block_size=110 io.logger=csv
37-
38-
coverage combine
39-
coverage xml -i

0 commit comments

Comments
 (0)