Update amd docker image to sglang:v0.4.5.post3-rocm630. (#5697)
This commit is contained in:
12
.github/workflows/pr-test-amd.yml
vendored
12
.github/workflows/pr-test-amd.yml
vendored
@@ -38,12 +38,12 @@ jobs:
|
|||||||
else
|
else
|
||||||
DEVICE_FLAG="--device /dev/dri"
|
DEVICE_FLAG="--device /dev/dri"
|
||||||
fi
|
fi
|
||||||
docker pull lmsysorg/sglang:v0.4.5.post2-rocm630
|
docker pull lmsysorg/sglang:v0.4.5.post3-rocm630
|
||||||
docker run -dt --user root --device=/dev/kfd $DEVICE_FLAG \
|
docker run -dt --user root --device=/dev/kfd $DEVICE_FLAG \
|
||||||
-v ${{ github.workspace }}:/sglang-checkout --ipc=host --group-add video \
|
-v ${{ github.workspace }}:/sglang-checkout --ipc=host --group-add video \
|
||||||
--cap-add=SYS_PTRACE -e HF_TOKEN=${HF_TOKEN} --security-opt seccomp=unconfined \
|
--cap-add=SYS_PTRACE -e HF_TOKEN=${HF_TOKEN} --security-opt seccomp=unconfined \
|
||||||
-w /sglang-checkout --name ci_sglang \
|
-w /sglang-checkout --name ci_sglang \
|
||||||
lmsysorg/sglang:v0.4.5.post2-rocm630
|
lmsysorg/sglang:v0.4.5.post3-rocm630
|
||||||
|
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
run: |
|
run: |
|
||||||
@@ -82,12 +82,12 @@ jobs:
|
|||||||
else
|
else
|
||||||
DEVICE_FLAG="--device /dev/dri"
|
DEVICE_FLAG="--device /dev/dri"
|
||||||
fi
|
fi
|
||||||
docker pull lmsysorg/sglang:v0.4.5.post2-rocm630
|
docker pull lmsysorg/sglang:v0.4.5.post3-rocm630
|
||||||
docker run -dt --user root --device=/dev/kfd $DEVICE_FLAG \
|
docker run -dt --user root --device=/dev/kfd $DEVICE_FLAG \
|
||||||
-v ${{ github.workspace }}:/sglang-checkout --ipc=host --group-add video \
|
-v ${{ github.workspace }}:/sglang-checkout --ipc=host --group-add video \
|
||||||
--cap-add=SYS_PTRACE -e HF_TOKEN=${{ secrets.AMD_HF_TOKEN }} --security-opt seccomp=unconfined \
|
--cap-add=SYS_PTRACE -e HF_TOKEN=${{ secrets.AMD_HF_TOKEN }} --security-opt seccomp=unconfined \
|
||||||
-w /sglang-checkout --name ci_sglang \
|
-w /sglang-checkout --name ci_sglang \
|
||||||
lmsysorg/sglang:v0.4.5.post2-rocm630
|
lmsysorg/sglang:v0.4.5.post3-rocm630
|
||||||
|
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
run: |
|
run: |
|
||||||
@@ -120,12 +120,12 @@ jobs:
|
|||||||
else
|
else
|
||||||
DEVICE_FLAG="--device /dev/dri"
|
DEVICE_FLAG="--device /dev/dri"
|
||||||
fi
|
fi
|
||||||
docker pull lmsysorg/sglang:v0.4.5.post2-rocm630
|
docker pull lmsysorg/sglang:v0.4.5.post3-rocm630
|
||||||
docker run -dt --user root --device=/dev/kfd $DEVICE_FLAG \
|
docker run -dt --user root --device=/dev/kfd $DEVICE_FLAG \
|
||||||
-v ${{ github.workspace }}:/sglang-checkout --ipc=host --group-add video \
|
-v ${{ github.workspace }}:/sglang-checkout --ipc=host --group-add video \
|
||||||
--cap-add=SYS_PTRACE -e HF_TOKEN=${HF_TOKEN} --security-opt seccomp=unconfined \
|
--cap-add=SYS_PTRACE -e HF_TOKEN=${HF_TOKEN} --security-opt seccomp=unconfined \
|
||||||
-w /sglang-checkout --name ci_sglang \
|
-w /sglang-checkout --name ci_sglang \
|
||||||
lmsysorg/sglang:v0.4.5.post2-rocm630
|
lmsysorg/sglang:v0.4.5.post3-rocm630
|
||||||
|
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
run: |
|
run: |
|
||||||
|
|||||||
Reference in New Issue
Block a user