Pin torch version until deprecated functionality is fixed (iree-org#147)

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]> Signed-off-by: Ian <[email protected]>
IanNod · Sep 30, 2024 · 9978a9a · 9978a9a
1 parent bae8e10
commit 9978a9a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pytorch-cpu-requirements.txt b/pytorch-cpu-requirements.txt
@@ -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
diff --git a/pytorch-rocm-requirements.txt b/pytorch-rocm-requirements.txt
@@ -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