Skip to content

Commit 5c51a6d

Browse files
authored
Fix broken extra 'cpu' (#5050)
1 parent 0b92b6c commit 5c51a6d

File tree

2 files changed

+81
-154
lines changed

2 files changed

+81
-154
lines changed

library/pyproject.toml

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -84,18 +84,18 @@ lint = [
8484

8585
[project.optional-dependencies]
8686
cpu = [
87-
"torch==2.9.1",
88-
"torchvision==0.24.1"
87+
"torch==2.9.1+cpu",
88+
"torchvision==0.24.1+cpu"
8989
]
9090
cuda = [
91-
"torch==2.9.1",
92-
"torchvision==0.24.1"
91+
"torch==2.9.1+cu128",
92+
"torchvision==0.24.1+cu128"
9393
]
9494
xpu = [
9595
"torch==2.9.0+xpu", # torch 2.9.1 with torchvision==0.24.1 is currently
9696
# unstable resulting in an error with torchvision ops.
9797
"pytorch-triton-xpu==3.5.0",
98-
"torchvision==0.24.0"
98+
"torchvision==0.24.0+xpu"
9999
]
100100

101101
docs = [

0 commit comments

Comments
 (0)