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", "blobfile==3.0.0",
"build", "build",
"compressed-tensors", "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')", "decord2 ; sys_platform == 'linux' and (platform_machine == 'aarch64' or platform_machine == 'arm64' or platform_machine == 'armv7l')",
"datasets", "datasets",
"einops", "einops",
"fastapi", "fastapi",
"flashinfer_python==0.6.6", # keep it aligned with jit-cache version in Dockerfile "flashinfer_python==0.6.12", # keep it aligned with jit-cache version in Dockerfile
"flashinfer_cubin==0.6.6", "flashinfer_cubin==0.6.12",
"gguf", "gguf",
"interegular", "interegular",
"llguidance>=0.7.11,<0.8.0", "llguidance>=0.7.11,<0.8.0",
@@ -58,17 +58,10 @@ dependencies = [
"sentencepiece", "sentencepiece",
"setproctitle", "setproctitle",
"flash-attn-4>=4.0.0b4", "flash-attn-4>=4.0.0b4",
"sglang-kernel==0.4.0", "sglang-kernel==0.4.3",
"soundfile==0.13.1", "soundfile==0.13.1",
"tiktoken", "tiktoken",
"timm==1.0.16", "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", "tqdm",
"mistral_common>=1.9.0", "mistral_common>=1.9.0",
"transformers==5.3.0", "transformers==5.3.0",