requirements: use stable torch packages on s390x (#26864)

This commit is contained in:
Niklas Wenzel
2026-08-11 21:58:53 +08:00
committed by GitHub
parent 5d16e81dd9
commit 2468576f24
4 changed files with 3 additions and 15 deletions
+1 -7
View File
@@ -2,11 +2,5 @@
--extra-index-url https://download.pytorch.org/whl/cpu
pillow~=11.3.0
## Embedding Gemma requires PyTorch 2.6.0 or later, bumped to 2.11.0 for compatibility
torch==2.11.0; platform_machine != "s390x" # check_requirements: ignore "=="
torch==2.11.0 # check_requirements: ignore "=="
torchvision==0.26.0; platform_machine != "s390x" # check_requirements: ignore "=="
# torch s390x packages can only be found from nightly builds
--extra-index-url https://download.pytorch.org/whl/nightly
torch>=0.0.0.dev0; platform_machine == "s390x" # check_requirements: ignore "=="
torchvision>=0.0.0.dev0; platform_machine == "s390x" # check_requirements: ignore "=="