diff --git a/docker/Dockerfile b/docker/Dockerfile index 3d35acba1..74319e82b 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -12,7 +12,7 @@ ARG DEEPEP_COMMIT=9af0e0d0e74f3577af1979c9b9e1ac2cad0104ee ARG BUILD_AND_DOWNLOAD_PARALLEL=8 ARG SGL_KERNEL_VERSION=0.4.4 ARG SGL_VERSION -ARG SGL_DEEP_GEMM_VERSION=0.1.3 +ARG SGL_DEEP_GEMM_VERSION=0.1.4.post1 ARG USE_LATEST_SGLANG=0 ARG GDRCOPY_VERSION=2.5.1 ARG PIP_DEFAULT_INDEX diff --git a/python/pyproject.toml b/python/pyproject.toml index 0b38dd679..0bcd942ff 100755 --- a/python/pyproject.toml +++ b/python/pyproject.toml @@ -64,7 +64,7 @@ dependencies = [ "scipy", "sentencepiece", "setproctitle", - "sgl-deep-gemm==0.1.4", + "sgl-deep-gemm==0.1.4.post1", "sglang-kernel==0.4.4", "smg-grpc-servicer>=0.5.0", "soundfile==0.13.1",