chore: update Dockerfile (#5894)

This commit is contained in:
Yineng Zhang
2025-04-29 12:55:13 -07:00
committed by GitHub
parent 771669cbe0
commit f4c191a712
2 changed files with 8 additions and 11 deletions

View File

@@ -14,8 +14,8 @@ jobs:
environment: 'prod'
strategy:
matrix:
cuda_version: ['11.8.0', '12.1.1', '12.4.1', '12.5.1']
build_type: ['all', 'srt']
cuda_version: ['12.4.1']
build_type: ['all']
steps:
- name: Delete huge unnecessary tools folder
run: rm -rf /opt/hostedtoolcache
@@ -60,7 +60,7 @@ jobs:
docker build . -f docker/Dockerfile --build-arg CUDA_VERSION=${{ matrix.cuda_version }} --build-arg BUILD_TYPE=${{ matrix.build_type }} -t lmsysorg/sglang:${tag}${tag_suffix} --no-cache
docker push lmsysorg/sglang:${tag}${tag_suffix}
if [ "${{ matrix.cuda_version }}" = "12.5.1" ]; then
if [ "${{ matrix.cuda_version }}" = "12.4.1" ]; then
docker tag lmsysorg/sglang:${tag}${tag_suffix} lmsysorg/sglang:latest${tag_suffix}
docker push lmsysorg/sglang:latest${tag_suffix}
fi