Update deps for mllama4 (#5215)

2025-04-11 00:12:44 +08:00
parent 7e4f72dd8c
commit 1078396f47
3 changed files with 26 additions and 25 deletions
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -38,7 +38,7 @@ runtime_common = [
    "pyzmq>=25.1.2",
    "soundfile==0.13.1",
    "torchao>=0.7.0",
-    "transformers==4.51.0",
+    "transformers==4.51.1",
    "uvicorn",
    "uvloop",
    "compressed-tensors",
@@ -50,6 +50,7 @@ srt = [
    "sgl-kernel==0.0.8",
    "flashinfer_python==0.2.3",
    "torch==2.5.1",
+    "torchvision==0.20.1",
    "cuda-python",
    "outlines>=0.0.44,<=0.1.11",
    "partial_json_parser",