chore: update torch v2.5.1 (#1849)

This commit is contained in:
Yineng Zhang
2024-11-18 00:06:00 +08:00
committed by GitHub
parent f719d9aebc
commit 3b878863f7
10 changed files with 174 additions and 37 deletions

View File

@@ -20,7 +20,7 @@ runtime_common = ["aiohttp", "decord", "fastapi", "hf_transfer", "huggingface_hu
"orjson", "packaging", "pillow", "prometheus-client>=0.20.0", "psutil", "pydantic", "python-multipart",
"torchao", "uvicorn", "uvloop", "pyzmq>=25.1.2",
"outlines>=0.0.44,<0.1.0", "modelscope"]
srt = ["sglang[runtime_common]", "torch", "vllm==0.6.3.post1"]
srt = ["sglang[runtime_common]", "torch", "vllm==0.6.4.post1"]
# HIP (Heterogeneous-computing Interface for Portability) for AMD
# => base docker rocm/vllm-dev:20241022, not from public vllm whl