Skip to content

Commit

Permalink
Pin torch version until deprecated functionality is fixed (iree-org#147)
Browse files Browse the repository at this point in the history
Recent torch 2.5 versions failing with `ImportError: cannot import name
'dynamic_dim' from 'torch.export'`, pin torch version `<2.5.0` for now,
until proper fix is ready
iree-org#145

---------

Signed-off-by: Ivan Butygin <[email protected]>
  • Loading branch information
Hardcode84 authored and IanNod committed Sep 30, 2024
1 parent 0392be8 commit c79681e
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion pytorch-cpu-requirements.txt
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
--pre
--index-url https://download.pytorch.org/whl/test/cpu
torch>=2.3.0
torch>=2.3.0, <2.5.0
torchaudio
torchvision
2 changes: 1 addition & 1 deletion pytorch-rocm-requirements.txt
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
--pre
--index-url https://download.pytorch.org/whl/nightly/rocm6.0
torch>=2.3.0
torch>=2.3.0, <2.5.0
torchaudio
torchvision

0 comments on commit c79681e

Please sign in to comment.