Skip to content

V
verl
Unverified Commit 59643585 by Chi Zhang Committed by GitHub
Options

[ci] fix: fix qwen0.5b megatron ci (#396)

parent 2440aa69
Showing with 2 additions and 0 deletions
tests/e2e/run_qwen_megatron.sh
......@@ -4,6 +4,8 @@ set -x
huggingface-cli download Qwen/Qwen2.5-0.5B
export VLLM_ATTENTION_BACKEND=XFORMERS
python3 -m verl.trainer.main_ppo --config-path=config \
--config-name='ppo_megatron_trainer.yaml'\
data.train_files=$HOME/data/gsm8k/train.parquet \
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment