From 386e55bef6b75e4527988d6a48cde9ccde23e050 Mon Sep 17 00:00:00 2001 From: Aaron Teo Date: Tue, 20 May 2025 22:34:47 +0800 Subject: [PATCH 1/2] common: update requirements.txt to include pytorch nightly for s390x Signed-off-by: Aaron Teo --- requirements/requirements-convert_hf_to_gguf.txt | 2 ++ requirements/requirements-convert_hf_to_gguf_update.txt | 2 ++ requirements/requirements-convert_lora_to_gguf.txt | 2 ++ 3 files changed, 6 insertions(+) diff --git a/requirements/requirements-convert_hf_to_gguf.txt b/requirements/requirements-convert_hf_to_gguf.txt index 8cb9c354f0152..f146481f189a2 100644 --- a/requirements/requirements-convert_hf_to_gguf.txt +++ b/requirements/requirements-convert_hf_to_gguf.txt @@ -1,3 +1,5 @@ -r ./requirements-convert_legacy_llama.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 torch~=2.2.1 diff --git a/requirements/requirements-convert_hf_to_gguf_update.txt b/requirements/requirements-convert_hf_to_gguf_update.txt index 8cb9c354f0152..f146481f189a2 100644 --- a/requirements/requirements-convert_hf_to_gguf_update.txt +++ b/requirements/requirements-convert_hf_to_gguf_update.txt @@ -1,3 +1,5 @@ -r ./requirements-convert_legacy_llama.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 torch~=2.2.1 diff --git a/requirements/requirements-convert_lora_to_gguf.txt b/requirements/requirements-convert_lora_to_gguf.txt index 5758076c41dc1..d091d564846bf 100644 --- a/requirements/requirements-convert_lora_to_gguf.txt +++ b/requirements/requirements-convert_lora_to_gguf.txt @@ -1,2 +1,4 @@ -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 From c8e52b34984a475e790be1351c80258ebd383b98 Mon Sep 17 00:00:00 2001 From: Aaron Teo Date: Thu, 22 May 2025 14:03:30 +0800 Subject: [PATCH 2/2] common: fix torch installation via pip for s390x Signed-off-by: Aaron Teo --- requirements/requirements-convert_hf_to_gguf.txt | 4 +++- requirements/requirements-convert_hf_to_gguf_update.txt | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/requirements/requirements-convert_hf_to_gguf.txt b/requirements/requirements-convert_hf_to_gguf.txt index f146481f189a2..431c596c12354 100644 --- a/requirements/requirements-convert_hf_to_gguf.txt +++ b/requirements/requirements-convert_hf_to_gguf.txt @@ -1,5 +1,7 @@ -r ./requirements-convert_legacy_llama.txt --extra-index-url https://download.pytorch.org/whl/cpu +torch~=2.2.1; platform_machine != "s390x" + # torch s390x packages can only be found from nightly builds --extra-index-url https://download.pytorch.org/whl/nightly -torch~=2.2.1 +torch>=0.0.0.dev0; platform_machine == "s390x" diff --git a/requirements/requirements-convert_hf_to_gguf_update.txt b/requirements/requirements-convert_hf_to_gguf_update.txt index f146481f189a2..431c596c12354 100644 --- a/requirements/requirements-convert_hf_to_gguf_update.txt +++ b/requirements/requirements-convert_hf_to_gguf_update.txt @@ -1,5 +1,7 @@ -r ./requirements-convert_legacy_llama.txt --extra-index-url https://download.pytorch.org/whl/cpu +torch~=2.2.1; platform_machine != "s390x" + # torch s390x packages can only be found from nightly builds --extra-index-url https://download.pytorch.org/whl/nightly -torch~=2.2.1 +torch>=0.0.0.dev0; platform_machine == "s390x"