diff --git a/docker/Dockerfile b/docker/Dockerfile index b3e352d41..d7f4ead45 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -19,7 +19,7 @@ ARG PIP_DEFAULT_INDEX ARG UBUNTU_MIRROR ARG GITHUB_ARTIFACTORY=github.com ARG INSTALL_FLASHINFER_JIT_CACHE=0 -ARG FLASHINFER_VERSION=0.6.7 +ARG FLASHINFER_VERSION=0.6.7.post2 ARG MOONCAKE_VERSION=0.3.9 #if need other arg please add in MOONCAKE_COMPILE_ARG ARG MOONCAKE_COMPILE_ARG="-DUSE_HTTP=ON -DUSE_MNNVL=ON -DUSE_CUDA=ON -DWITH_EP=ON" diff --git a/python/pyproject.toml b/python/pyproject.toml index 5ca83a688..d388d275d 100755 --- a/python/pyproject.toml +++ b/python/pyproject.toml @@ -27,8 +27,8 @@ dependencies = [ "datasets", "einops", "fastapi", - "flashinfer_python==0.6.7", # keep it aligned with jit-cache version in Dockerfile - "flashinfer_cubin==0.6.7", + "flashinfer_python==0.6.7.post2", # keep it aligned with jit-cache version in Dockerfile + "flashinfer_cubin==0.6.7.post2", "gguf", "interegular", "llguidance>=0.7.11,<0.8.0", diff --git a/python/sglang/srt/entrypoints/engine.py b/python/sglang/srt/entrypoints/engine.py index 96769a14c..aca1adc26 100644 --- a/python/sglang/srt/entrypoints/engine.py +++ b/python/sglang/srt/entrypoints/engine.py @@ -1128,7 +1128,7 @@ def _set_envs_and_config(server_args: ServerArgs): if server_args.attention_backend == "flashinfer": assert_pkg_version( "flashinfer_python", - "0.6.7", + "0.6.7.post2", "Please uninstall the old version and " "reinstall the latest version by following the instructions " "at https://docs.flashinfer.ai/installation.html.", diff --git a/python/sglang/srt/utils/common.py b/python/sglang/srt/utils/common.py index 237c2dc44..0e274b78b 100644 --- a/python/sglang/srt/utils/common.py +++ b/python/sglang/srt/utils/common.py @@ -1040,7 +1040,7 @@ def check_pkg_version_at_least(pkg: str, min_version: str) -> bool: Args: pkg: Package name (distribution name, e.g., "flashinfer-python") - min_version: Minimum version required (e.g., "0.6.7") + min_version: Minimum version required (e.g., "0.6.7.post2") Returns: True if package is installed and version >= min_version, False otherwise