Skip to content

GitLab

Unverified Commit a968c888 authored by Wenxuan Tan's avatar Wenxuan Tan Committed by GitHub
Browse files

Fix torchvision version for Blackwell (#7015)

parent a979daac
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python/pyproject.toml
View file @ a968c888
...@@ -64,7 +64,7 @@ blackwell = [ ...@@ -64,7 +64,7 @@ blackwell = [
"sgl-kernel", "sgl-kernel",
"torch==2.7.1", "torch==2.7.1",
"torchaudio==2.7.1", "torchaudio==2.7.1",
"torchvision==0.22.0", "torchvision==0.22.1",
"cuda-python", "cuda-python",
"outlines>=0.0.44,<=0.1.11", "outlines>=0.0.44,<=0.1.11",
"einops", "einops",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment