From a71a44f20369384c986a99836af25d1b302653af Mon Sep 17 00:00:00 2001 From: Lianmin Zheng Date: Thu, 7 Nov 2024 19:20:47 -0800 Subject: [PATCH] Update setup_github_runner.md (#1952) --- docs/developer/setup_github_runner.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/developer/setup_github_runner.md b/docs/developer/setup_github_runner.md index c6218a0b78..8dad7b8c06 100644 --- a/docs/developer/setup_github_runner.md +++ b/docs/developer/setup_github_runner.md @@ -1,4 +1,4 @@ -# Set Up Self-hosted Runners for GitHub Action +# Set Up Self-Hosted Runners for GitHub Action ## Add a Runner @@ -9,9 +9,9 @@ You can mount a folder for the shared huggingface model weights cache. The comma ``` docker pull nvidia/cuda:12.1.1-devel-ubuntu22.04 # Nvidia -docker run --shm-size 64g -it -v /tmp/huggingface:/hf_home --gpus all nvidia/cuda:12.1.1-devel-ubuntu22.04 /bin/bash +docker run --shm-size 128g -it -v /tmp/huggingface:/hf_home --gpus all nvidia/cuda:12.1.1-devel-ubuntu22.04 /bin/bash # AMD -docker run --rm --device=/dev/kfd --device=/dev/dri --group-add video --shm-size 64g -it -v /tmp/huggingface:/hf_home henryx/haisgl:sgl0.3.1.post3_vllm0.6.0_triton3.0.0_rocm6.2.1 /bin/bash +docker run --rm --device=/dev/kfd --device=/dev/dri --group-add video --shm-size 128g -it -v /tmp/huggingface:/hf_home henryx/haisgl:sgl0.3.1.post3_vllm0.6.0_triton3.0.0_rocm6.2.1 /bin/bash ``` ### Step 2: Configure the runner by `config.sh`