Update requirements.txt
Browse files- requirements.txt +8 -5
requirements.txt
CHANGED
|
@@ -1,15 +1,18 @@
|
|
| 1 |
-
|
|
|
|
|
|
|
|
|
|
| 2 |
git+https://github.com/huggingface/accelerate.git
|
| 3 |
-
bitsandbytes
|
| 4 |
git+https://github.com/huggingface/transformers.git
|
| 5 |
-
git+https://github.com/huggingface/optimum.git
|
| 6 |
huggingface_hub
|
| 7 |
sentencepiece
|
|
|
|
| 8 |
paramiko
|
| 9 |
gradio
|
| 10 |
numpy
|
| 11 |
opencv-python
|
| 12 |
git+https://github.com/asomoza/image_gen_aux.git
|
| 13 |
git+https://github.com/ford442/diffusers.git@offload_sd3_del
|
| 14 |
-
peft
|
| 15 |
-
#flash-attn
|
|
|
|
| 1 |
+
--extra-index-url https://download.pytorch.org/whl/cu124
|
| 2 |
+
torch
|
| 3 |
+
torchvision
|
| 4 |
+
torchaudio
|
| 5 |
git+https://github.com/huggingface/accelerate.git
|
| 6 |
+
#bitsandbytes
|
| 7 |
git+https://github.com/huggingface/transformers.git
|
| 8 |
+
#git+https://github.com/huggingface/optimum.git
|
| 9 |
huggingface_hub
|
| 10 |
sentencepiece
|
| 11 |
+
safetensors
|
| 12 |
paramiko
|
| 13 |
gradio
|
| 14 |
numpy
|
| 15 |
opencv-python
|
| 16 |
git+https://github.com/asomoza/image_gen_aux.git
|
| 17 |
git+https://github.com/ford442/diffusers.git@offload_sd3_del
|
| 18 |
+
peft
|
|
|