From 78706ed9a833b7e6651c8ff4efc09e7ba31e0fed Mon Sep 17 00:00:00 2001 From: compilade Date: Sun, 7 Jul 2024 07:23:38 -0400 Subject: [PATCH] py : use cpu-only torch in requirements.txt (#8335) --- examples/llava/requirements.txt | 1 + requirements/requirements-convert_hf_to_gguf.txt | 1 + requirements/requirements-convert_hf_to_gguf_update.txt | 1 + 3 files changed, 3 insertions(+) diff --git a/examples/llava/requirements.txt b/examples/llava/requirements.txt index 4713f0a3460f4..fbfd0cdd722e1 100644 --- a/examples/llava/requirements.txt +++ b/examples/llava/requirements.txt @@ -1,3 +1,4 @@ -r ../../requirements/requirements-convert_legacy_llama.txt +--extra-index-url https://download.pytorch.org/whl/cpu pillow~=10.2.0 torch~=2.2.1 diff --git a/requirements/requirements-convert_hf_to_gguf.txt b/requirements/requirements-convert_hf_to_gguf.txt index 653355c07d68c..8cb9c354f0152 100644 --- a/requirements/requirements-convert_hf_to_gguf.txt +++ b/requirements/requirements-convert_hf_to_gguf.txt @@ -1,2 +1,3 @@ -r ./requirements-convert_legacy_llama.txt +--extra-index-url https://download.pytorch.org/whl/cpu 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 653355c07d68c..8cb9c354f0152 100644 --- a/requirements/requirements-convert_hf_to_gguf_update.txt +++ b/requirements/requirements-convert_hf_to_gguf_update.txt @@ -1,2 +1,3 @@ -r ./requirements-convert_legacy_llama.txt +--extra-index-url https://download.pytorch.org/whl/cpu torch~=2.2.1