From 3605067898f7257f691a5c7a2aee2f9ea1c7ce5f Mon Sep 17 00:00:00 2001 From: kingbri Date: Mon, 29 Jan 2024 23:30:10 -0500 Subject: [PATCH] Requirements: Don't use torch 2.2 Pytorch released 2.2 without letting the community know first. Pin the torch version to 2.1.2 until exllamav2 builds for torch 2.2 Signed-off-by: kingbri --- requirements-amd.txt | 3 ++- requirements-cu118.txt | 3 ++- requirements.txt | 3 ++- 3 files changed, 6 insertions(+), 3 deletions(-) diff --git a/requirements-amd.txt b/requirements-amd.txt index 8a68e72..416e897 100644 --- a/requirements-amd.txt +++ b/requirements-amd.txt @@ -1,6 +1,7 @@ # Torch +# Patch until torch 2.2 is supported --extra-index-url https://download.pytorch.org/whl/rocm5.6 -torch +torch < 2.2, >= 2 # Exllamav2 https://github.com/turboderp/exllamav2/releases/download/v0.0.12/exllamav2-0.0.12+rocm5.6-cp311-cp311-linux_x86_64.whl; python_version == "3.11" diff --git a/requirements-cu118.txt b/requirements-cu118.txt index f176bac..ed0042b 100644 --- a/requirements-cu118.txt +++ b/requirements-cu118.txt @@ -1,6 +1,7 @@ # Torch +# Patch until torch 2.2 is supported --extra-index-url https://download.pytorch.org/whl/cu118 -torch +torch < 2.2, >= 2 # Exllamav2 diff --git a/requirements.txt b/requirements.txt index 136715b..4e529e1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,7 @@ # Torch +# Patch until torch 2.2 is supported --extra-index-url https://download.pytorch.org/whl/cu121 -torch +torch < 2.2, >= 2 # Exllamav2