Skip to content

Commit

Permalink
Switch torch requirements back to stable channel. (#347)
Browse files Browse the repository at this point in the history
We have some test failures with 2.6.0 prereleases:
https://github.com/iree-org/iree-turbine/actions/runs/12399117695?pr=346.
Sticking to stable versions until we can triage.
  • Loading branch information
ScottTodd authored Dec 18, 2024
1 parent 08b5752 commit ffdd592
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 3 deletions.
3 changes: 1 addition & 2 deletions pytorch-cpu-requirements.txt
Original file line number Diff line number Diff line change
@@ -1,5 +1,4 @@
--pre
--index-url https://download.pytorch.org/whl/test/cpu
--index-url https://download.pytorch.org/whl/cpu
torch>=2.3.0
torchaudio
torchvision
1 change: 0 additions & 1 deletion pytorch-rocm-requirements.txt
Original file line number Diff line number Diff line change
@@ -1,4 +1,3 @@
--pre
--index-url https://download.pytorch.org/whl/rocm6.2
torch>=2.3.0
torchaudio
Expand Down

0 comments on commit ffdd592

Please sign in to comment.