From c8c736e75ce9be74fc0e76c8f25860c423c81aa5 Mon Sep 17 00:00:00 2001
From: laoyao0822 <laoyao0822@qq.com>
Date: Wed, 10 Jun 2026 05:54:43 +0800
Subject: [PATCH] new pyprojectoml

---
 python/pyproject.toml | 15 ++++-----------
 1 file changed, 4 insertions(+), 11 deletions(-)

diff --git a/python/pyproject.toml b/python/pyproject.toml
index 5bdd73ae6..b967634e6 100755
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -22,13 +22,13 @@ dependencies = [
   "blobfile==3.0.0",
   "build",
   "compressed-tensors",
-  "cuda-python==12.9",
+  "cuda-python==13.2.0",
   "decord2 ; sys_platform == 'linux' and (platform_machine == 'aarch64' or platform_machine == 'arm64' or platform_machine == 'armv7l')",
   "datasets",
   "einops",
   "fastapi",
-  "flashinfer_python==0.6.6", # keep it aligned with jit-cache version in Dockerfile
-  "flashinfer_cubin==0.6.6",
+  "flashinfer_python==0.6.12", # keep it aligned with jit-cache version in Dockerfile
+  "flashinfer_cubin==0.6.12",
   "gguf",
   "interegular",
   "llguidance>=0.7.11,<0.8.0",
@@ -58,17 +58,10 @@ dependencies = [
   "sentencepiece",
   "setproctitle",
   "flash-attn-4>=4.0.0b4",
-  "sglang-kernel==0.4.0",
+  "sglang-kernel==0.4.3",
   "soundfile==0.13.1",
   "tiktoken",
   "timm==1.0.16",
-  "torch_memory_saver==0.0.9",
-  "torch==2.9.1",
-  "torchao==0.9.0",
-  "torchaudio==2.9.1",
-  "torchcodec==0.9.1 ; sys_platform != 'linux' or (sys_platform == 'linux' and platform_machine != 'aarch64' and platform_machine != 'arm64' and platform_machine != 'armv7l')", # torchcodec 0.9.1 for torch 2.9.x. Not available on Linux ARM.
-  "av ; sys_platform == 'linux' and (platform_machine == 'aarch64' or platform_machine == 'arm64' or platform_machine == 'armv7l')",
-  "torchvision",
   "tqdm",
   "mistral_common>=1.9.0",
   "transformers==5.3.0",