diff --git a/python/pyproject.toml b/python/pyproject.toml index 89267f397..2650861a8 100755 --- a/python/pyproject.toml +++ b/python/pyproject.toml @@ -33,7 +33,7 @@ dependencies = [ "flash-attn-4>=4.0.0b18", "flashinfer_python[cu13]==0.6.15.post1", # keep it aligned with jit-cache version in Dockerfile "gguf", - "helion==0.2.6", + "helion==1.4", "humming-kernels[cu13]==0.1.10", "interegular", "IPython", diff --git a/python/pyproject_other.toml b/python/pyproject_other.toml index 61aeb106e..fa7ee033a 100755 --- a/python/pyproject_other.toml +++ b/python/pyproject_other.toml @@ -30,7 +30,7 @@ runtime_common = [ "einops", "fastapi", "gguf", - "helion==0.2.6", + "helion==1.4", "interegular", "IPython", "llguidance>=1.7.6,<2.0.0",