match torch and torch_vision version (#2649)

InternLM · Oct 24, 2024 · 89f52bc · 89f52bc
1 parent 4958071
commit 89f52bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements/runtime.txt b/requirements/runtime.txt
@@ -16,7 +16,7 @@ sentencepiece
 shortuuid
 tiktoken
 torch<=2.4.0,>=2.0.0
-torchvision<=0.18.1,>=0.15.0
+torchvision<=0.19.0,>=0.15.0
 transformers
 triton>=2.2.0,<=3.0.0; sys_platform == "linux"
 uvicorn