Skip to content

Commit ec73b9c

Browse files
committed
require torch 2.8 to pass onnx export tests
1 parent a96f677 commit ec73b9c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ dev = [
7070
"segment-anything", # for model testing
7171
"tensorflow",
7272
"timm", # for model testing
73-
"torch>=1.6,<3",
73+
"torch>=2.8,<3",
7474
"torchvision>=0.21",
7575
]
7676

0 commit comments

Comments
 (0)