Skip to content

Commit 0820e93

Browse files
authored
[torchlib] Fix various implementations (#2050)
Fix implementations according to pytorch/pytorch#146224. Removed Eager Mode tests since we only care about the graph constructed. Changed all traceable ops to trace_only. Removed usage of the IsScalar function.
1 parent 89d3446 commit 0820e93

File tree

12 files changed

+371
-561
lines changed

12 files changed

+371
-561
lines changed

noxfile.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,8 @@
3232
)
3333
ONNX = "onnx==1.17"
3434
ONNX_RUNTIME = "onnxruntime==1.20.1"
35-
PYTORCH = "torch==2.4.1"
36-
TORCHVISON = "torchvision==0.19.1"
35+
PYTORCH = "torch==2.5.1"
36+
TORCHVISON = "torchvision==0.20.1"
3737
TRANSFORMERS = "transformers==4.37.2"
3838
ONNX_RUNTIME_NIGHTLY_DEPENDENCIES = (
3939
"flatbuffers",

0 commit comments

Comments
 (0)