Skip to content

Commit

Permalink
Change install dependencies in cpu-test.
Browse files Browse the repository at this point in the history
  • Loading branch information
jmaksymczuk committed Sep 3, 2024
1 parent 6ec20c3 commit 0bc0494
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions .github/workflows/cpu-test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -26,9 +26,7 @@ jobs:
- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip install torch --extra-index-url https://download.pytorch.org/whl/cpu
pip install -r requirements-hpu.txt
VLLM_TARGET_DEVICE=hpu python setup.py develop
pip install -e .
- name: cpu-test
run: |
VLLM_SKIP_WARMUP=true VLLM_PROMPT_SEQ_BUCKET_MAX=128 python examples/offline_inference_fakehpu.py --fake_hpu

0 comments on commit 0bc0494

Please sign in to comment.