Update requirements.txt
Browse files- requirements.txt +11 -8
requirements.txt
CHANGED
|
@@ -1,9 +1,12 @@
|
|
| 1 |
-
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
bitsandbytes
|
| 7 |
-
|
|
|
|
| 8 |
gradio[oauth]
|
| 9 |
-
tensorboard
|
|
|
|
|
|
|
|
|
| 1 |
+
torch>=2.1.0
|
| 2 |
+
accelerate>=0.25.0
|
| 3 |
+
xformers==0.0.22
|
| 4 |
+
trl==0.7.6
|
| 5 |
+
peft>=0.6.0
|
| 6 |
+
bitsandbytes>=0.41.1
|
| 7 |
+
unsloth @ git+https://github.com/unslothai/unsloth.git
|
| 8 |
+
gradio>=4.0.0
|
| 9 |
gradio[oauth]
|
| 10 |
+
tensorboard
|
| 11 |
+
transformers>=4.34.0
|
| 12 |
+
huggingface-hub>=0.19.0
|