theroyallab/tabbyAPI
1
0
Fork 0
mirror of https://github.com/theroyallab/tabbyAPI.git synced 2026-03-14 15:57:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Requirements: Fix torch install

Browse Source 
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 <bdashore3@proton.me>
This commit is contained in:
kingbri
2023-12-06 19:04:35 -05:00
parent f8e9e22c43
commit 47176a2a1e
3 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
requirements-amd.txt
View File

@@ -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

3
requirements-cu118.txt
View File

@@ -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

3
requirements.txt
View File

@@ -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