Update requirements.txt
Browse files- requirements.txt +2 -3
requirements.txt
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
--extra-index-url https://download.pytorch.org/whl/nightly/
|
2 |
torch>=2.0.1
|
3 |
git+https://github.com/huggingface/optimum.git
|
4 |
git+https://github.com/huggingface/transformers.git
|
@@ -8,5 +8,4 @@ bitsandbytes
|
|
8 |
accelerate
|
9 |
torchvision
|
10 |
requests
|
11 |
-
|
12 |
-
https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.4.1/auto_gptq-0.4.1+cu118-cp310-cp310-linux_x86_64.whl
|
|
|
1 |
+
--extra-index-url https://download.pytorch.org/whl/nightly/cu117
|
2 |
torch>=2.0.1
|
3 |
git+https://github.com/huggingface/optimum.git
|
4 |
git+https://github.com/huggingface/transformers.git
|
|
|
8 |
accelerate
|
9 |
torchvision
|
10 |
requests
|
11 |
+
https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.4.1/auto_gptq-0.4.1+cu117-cp39-cp39-linux_x86_64.whl
|
|