Spaces:
Runtime error
Runtime error
FlawedLLM
commited on
Update requirements.txt
Browse files- requirements.txt +12 -5
requirements.txt
CHANGED
@@ -1,6 +1,13 @@
|
|
1 |
transformers==4.41.0
|
2 |
-
|
3 |
-
|
4 |
-
|
5 |
-
|
6 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
transformers==4.41.0
|
2 |
+
# PyTorch and GPU Support
|
3 |
+
torch-cuda==12.1 # Or torch-cuda==11.8, depending on your exact needs
|
4 |
+
torchvision-cuda==0.13.1 # Ensure compatibility with PyTorch 12.1
|
5 |
+
cudatoolkit # No version specified; Conda will manage this
|
6 |
+
xformers # No version specified; Conda will manage this
|
7 |
+
|
8 |
+
# unsloth (and related)
|
9 |
+
unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git
|
10 |
+
trl
|
11 |
+
peft
|
12 |
+
accelerate
|
13 |
+
bitsandbytes
|