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
@@ -2,8 +2,4 @@
--extra-index-url https://download.pytorch.org/whl/cpu
## Embedding Gemma requires PyTorch 2.6.0 or later, bumped to 2.11.0 for compatibility
torch==2.11.0; platform_machine != "s390x"
# 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"
torch==2.11.0
@@ -1,4 +1,2 @@
-r ./requirements-convert_hf_to_gguf.txt
--extra-index-url https://download.pytorch.org/whl/cpu
# torch s390x packages can only be found from nightly builds
--extra-index-url https://download.pytorch.org/whl/nightly