Update requirements.txt
Browse files- requirements.txt +3 -0
requirements.txt
CHANGED
@@ -7,3 +7,6 @@ sentencepiece==0.1.99
|
|
7 |
torch==2.0.1
|
8 |
transformers==4.31.0
|
9 |
auto-gptq @ https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.3.2/auto_gptq-0.3.2+cu118-cp310-cp310-linux_x86_64.whl#sha256=5f05e44c440e9f72a0dba8b7f1e10d7ceacb041d21c68382702bee068eea3f77
|
|
|
|
|
|
|
|
7 |
torch==2.0.1
|
8 |
transformers==4.31.0
|
9 |
auto-gptq @ https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.3.2/auto_gptq-0.3.2+cu118-cp310-cp310-linux_x86_64.whl#sha256=5f05e44c440e9f72a0dba8b7f1e10d7ceacb041d21c68382702bee068eea3f77
|
10 |
+
|
11 |
+
--extra-index-url https://download.pytorch.org/whl/cu118
|
12 |
+
torch
|