diff --git a/pyproject.toml b/pyproject.toml index ff2ffd9a28..b9fafdd856 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -70,7 +70,7 @@ automodel = [ "causal-conv1d", "nv-grouped-gemm", "transformer-engine[pytorch]>=2.9.0a0,<2.12.0", - "deep_ep @ git+https://github.com/deepseek-ai/DeepEP.git@bfded34800dfec415b71503f8205181de90b2480", + "deep_ep @ git+https://github.com/deepseek-ai/DeepEP.git@b306af06afd412c88e51e71802951606e40b7358", ] vllm = [ "cuda-python", @@ -78,7 +78,7 @@ vllm = [ # deep_ep also needs libibverbs-dev # sudo apt-get update # sudo apt-get install libibverbs-dev - "deep_ep @ git+https://github.com/deepseek-ai/DeepEP.git@bfded34800dfec415b71503f8205181de90b2480", + "deep_ep @ git+https://github.com/deepseek-ai/DeepEP.git@b306af06afd412c88e51e71802951606e40b7358", "vllm==0.17.1", "num2words>=0.5.14", "flashinfer-python==0.6.4", @@ -107,7 +107,7 @@ mcore = [ # https://github.com/facebookresearch/xformers/blob/8354497deb2c04c67fbb2e2ad911e86530da0e90/xformers/ops/fmha/flash.py#L76 "flash-attn==2.8.1", "emerging-optimizers==0.2.0", - "deep_ep @ git+https://github.com/deepseek-ai/DeepEP.git@bfded34800dfec415b71503f8205181de90b2480", + "deep_ep @ git+https://github.com/deepseek-ai/DeepEP.git@b306af06afd412c88e51e71802951606e40b7358", ] modelopt = ["nvidia-modelopt"] nvrx = [ @@ -253,7 +253,7 @@ override-dependencies = [ "llguidance>=1.3.0,<1.4.0", # Override setuptools range in other dependencies to address CVE GHSA-58pv-8j8x-9vj2 "setuptools>=80.10.2", - "deep_ep @ git+https://github.com/deepseek-ai/DeepEP.git@bfded34800dfec415b71503f8205181de90b2480", + "deep_ep @ git+https://github.com/deepseek-ai/DeepEP.git@b306af06afd412c88e51e71802951606e40b7358", # Note: flashinfer versions are pinned per-extra (vllm uses 0.6.4, sglang uses 0.6.7.post2) # since vllm and sglang extras are mutually exclusive and have different requirements. # Override megatron-core's flashinfer~=0.5.0 constraint to allow both vllm (0.6.4) and sglang (0.6.7.post2)