diff --git a/.buildkite/run-amd-test.sh b/.buildkite/run-amd-test.sh index b459b1097e5fa..618d712b0279b 100644 --- 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} \