Use 16gb shm for container

vllm-project · Jul 19, 2024 · 39c2cc5 · 39c2cc5
1 parent 9dbfbc8
commit 39c2cc5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.buildkite/run-amd-test.sh b/.buildkite/run-amd-test.sh
@@ -73,6 +73,7 @@ HF_MOUNT="/root/.cache/huggingface"
 docker run \
         --device /dev/kfd --device /dev/dri \
         --network host \
+        --shm-size=16gb \
         --rm \
         -e HF_TOKEN \
         -v ${HF_CACHE}:${HF_MOUNT} \