Unverified Commit 594d80ad by Chi Zhang Committed by GitHub

[ci] fix: change VLLM_ATTENTION_BACKEND to XFORMERS to avoid illegal memory access (#113)

parent 018b0d73
......@@ -15,6 +15,7 @@ on:
paths:
- "**/*.py"
- .github/workflows/e2e_digit_completion.yml
- "tests/e2e/*.sh"
jobs:
e2e_digit_completion:
......
......@@ -15,6 +15,7 @@ on:
paths:
- "**/*.py"
- .github/workflows/e2e_gsm8k.yml
- "tests/e2e/*.sh"
jobs:
e2e_gsm8k:
......
set -x
export VLLM_ATTENTION_BACKEND=XFORMERS
python3 -m verl.trainer.main_ppo \
data.train_files=$HOME/data/gsm8k/train.parquet \
data.val_files=$HOME/data/gsm8k/test.parquet \
......
set -x
export VLLM_ATTENTION_BACKEND=XFORMERS
python3 -m verl.trainer.main_ppo \
data.train_files=$HOME/data/gsm8k/train.parquet \
data.val_files=$HOME/data/gsm8k/test.parquet \
......
set -x
export VLLM_ATTENTION_BACKEND=XFORMERS
python3 -m verl.trainer.main_ppo \
data.train_files=$HOME/data/gsm8k/train.parquet \
data.val_files=$HOME/data/gsm8k/test.parquet \
......
set -x
export VLLM_ATTENTION_BACKEND=XFORMERS
python3 -m verl.trainer.main_ppo \
data.train_files=$HOME/data/gsm8k/train.parquet \
data.val_files=$HOME/data/gsm8k/test.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