Update flashinfer to 0.6.1 (#15551)

2026-01-17 00:48:30 +08:00
parent 82a1b645ba
commit a04675892e
10 changed files with 6 additions and 11 deletions
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -28,8 +28,8 @@ dependencies = [
  "datasets",
  "einops",
  "fastapi",
-  "flashinfer_python==0.5.3", # keep it aligned with jit-cache version in Dockerfile
-  "flashinfer_cubin==0.5.3",
+  "flashinfer_python==0.6.1", # keep it aligned with jit-cache version in Dockerfile
+  "flashinfer_cubin==0.6.1",
  "gguf",
  "hf_transfer",
  "huggingface_hub",