Pin torch version until deprecated functionality is fixed (#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 #145 --------- Signed-off-by: Ivan Butygin <[email protected]>
iree-org · Sep 17, 2024 · 7300a8d · 7300a8d
1 parent bae8e10
commit 7300a8d
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