From 47176a2a1ea22f04b76af53bae336b6fe6081b2e Mon Sep 17 00:00:00 2001 From: kingbri Date: Wed, 6 Dec 2023 19:04:35 -0500 Subject: [PATCH] Requirements: Fix torch install Use --extra-index-url to install pytorch. This should be secure enough since dependency confusion attacks aren't possible with just installing the torch package. 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 42be584..928c322 100644 --- a/requirements-amd.txt +++ b/requirements-amd.txt @@ -1,5 +1,6 @@ # Torch -torch --index-url https://download.pytorch.org/whl/rocm5.6 +--extra-index-url https://download.pytorch.org/whl/rocm5.6 +torch # Exllamav2 https://github.com/turboderp/exllamav2/releases/download/v0.0.10/exllamav2-0.0.10+rocm5.6-cp311-cp311-linux_x86_64.whl diff --git a/requirements-cu118.txt b/requirements-cu118.txt index 563bf04..7b30510 100644 --- a/requirements-cu118.txt +++ b/requirements-cu118.txt @@ -1,5 +1,6 @@ # Torch -torch --index-url https://download.pytorch.org/whl/cu118 +--extra-index-url https://download.pytorch.org/whl/cu118 +torch # Exllamav2 diff --git a/requirements.txt b/requirements.txt index 1a93c08..be86943 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,6 @@ # Torch -torch --index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu121 +torch # Exllamav2