diff --git a/docker/Dockerfile b/docker/Dockerfile index 104d0f479..c1eeb6235 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.3 ARG SGL_VERSION -ARG SGL_DEEP_GEMM_VERSION=0.1.1 +ARG SGL_DEEP_GEMM_VERSION=0.1.2 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 9d1ca121b..863100d9b 100755 --- a/python/pyproject.toml +++ b/python/pyproject.toml @@ -60,7 +60,7 @@ dependencies = [ "sentencepiece", "setproctitle", "flash-attn-4>=4.0.0b9", - "sgl-deep-gemm==0.1.1", + "sgl-deep-gemm==0.1.2", "sglang-kernel==0.4.3", "soundfile==0.13.1", "tiktoken",