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

diff --git a/examples/model-conversion/requirements.txt b/examples/model-conversion/requirements.txt
index 229b2ec75b75b8fb0f564fa7b8c85fdae6ff8bd0..d2cd357ec95c6102a2228825619c15572344418b 100644 (file)
--- a/examples/model-conversion/requirements.txt
+++ b/examples/model-conversion/requirements.txt
@@ -1,6 +1,6 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
 torch
-torchvision
+torchvision; platform_machine != "s390x"
 transformers
 huggingface-hub
 accelerate

diff --git a/requirements/requirements-convert_hf_to_gguf.txt b/requirements/requirements-convert_hf_to_gguf.txt
index f80fdc1f6402c5c1652aa0a2fb1eada81772978a..b1f7c863e27e184e55408c9e5792158301c86547 100644 (file)
--- a/requirements/requirements-convert_hf_to_gguf.txt
+++ b/requirements/requirements-convert_hf_to_gguf.txt
@@ -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

diff --git a/requirements/requirements-convert_lora_to_gguf.txt b/requirements/requirements-convert_lora_to_gguf.txt
index d091d564846bff78972a8502baa6a57c672b7abe..5758076c41dc116c5d41fccb4cd60ef67afaa20e 100644 (file)
--- a/requirements/requirements-convert_lora_to_gguf.txt
+++ b/requirements/requirements-convert_lora_to_gguf.txt
@@ -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

diff --git a/tools/mtmd/requirements.txt b/tools/mtmd/requirements.txt
index f26d8e912a37d0b0d29a971f744817becb6a784b..d646ca7b02f2c2b49a37d5fccccd5e66e8269ad1 100644 (file)
--- a/tools/mtmd/requirements.txt
+++ b/tools/mtmd/requirements.txt
@@ -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 "=="