Commit
•
49c21e1
1
Parent(s):
68304b9
updated requirements
Browse files- requirements.txt +11 -1
requirements.txt
CHANGED
|
@@ -1 +1,11 @@
|
|
| 1 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
torch>=2.0.0
|
| 2 |
+
torchvision
|
| 3 |
+
transformers>=4.25.0
|
| 4 |
+
unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git
|
| 5 |
+
xformers==0.0.27
|
| 6 |
+
trl
|
| 7 |
+
peft
|
| 8 |
+
accelerate
|
| 9 |
+
bitsandbytes
|
| 10 |
+
triton
|
| 11 |
+
wandb
|