From 2468576f241235452013308597e6de1b78866996 Mon Sep 17 00:00:00 2001 From: Niklas Wenzel Date: Tue, 11 Aug 2026 15:58:53 +0200 Subject: [PATCH] requirements: use stable torch packages on s390x (#26864) --- examples/model-conversion/requirements.txt | 2 +- requirements/requirements-convert_hf_to_gguf.txt | 6 +----- requirements/requirements-convert_lora_to_gguf.txt | 2 -- tools/mtmd/requirements.txt | 8 +------- 4 files changed, 3 insertions(+), 15 deletions(-) diff --git a/examples/model-conversion/requirements.txt b/examples/model-conversion/requirements.txt index 229b2ec75b..d2cd357ec9 100644 --- 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 f80fdc1f64..b1f7c863e2 100644 --- 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 d091d56484..5758076c41 100644 --- 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 f26d8e912a..d646ca7b02 100644 --- 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 "=="