Skip to content

Commit

Permalink
[CI/Build] Reducing CPU CI execution time (vllm-project#5241)
Browse files Browse the repository at this point in the history
  • Loading branch information
bigPYJ1151 authored Jun 4, 2024
1 parent 9ba093b commit 45c35f0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion .buildkite/run-cpu-test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ trap remove_docker_container EXIT
remove_docker_container

# Run the image
docker run -itd -v ~/.cache/huggingface:/root/.cache/huggingface --network host -e HF_TOKEN --env VLLM_CPU_KVCACHE_SPACE=4 --name cpu-test cpu-test
docker run -itd -v ~/.cache/huggingface:/root/.cache/huggingface --cpuset-cpus=48-95 --cpuset-mems=1 --network host -e HF_TOKEN --env VLLM_CPU_KVCACHE_SPACE=4 --name cpu-test cpu-test

# offline inference
docker exec cpu-test bash -c "python3 examples/offline_inference.py"
Expand Down

0 comments on commit 45c35f0

Please sign in to comment.