new pyprojectoml

This commit is contained in:
laoyao0822
2026-06-10 05:54:43 +08:00
parent af20334f4c
commit c8c736e75c

View File

@@ -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",