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