Spaces:
Sleeping
Sleeping
Update requirements.txt
Browse files- requirements.txt +5 -19
requirements.txt
CHANGED
@@ -1,19 +1,5 @@
|
|
1 |
-
|
2 |
-
|
3 |
-
|
4 |
-
|
5 |
-
|
6 |
-
|
7 |
-
# Transformers and related packages
|
8 |
-
transformers>=4.36.0
|
9 |
-
peft>=0.7.0
|
10 |
-
accelerate>=0.25.0
|
11 |
-
huggingface_hub>=0.19.0
|
12 |
-
|
13 |
-
# BitsAndBytes with CUDA support
|
14 |
-
bitsandbytes>=0.41.0
|
15 |
-
|
16 |
-
# Additional dependencies for GPU support
|
17 |
-
ninja
|
18 |
-
flash-attn>=2.0.0 --no-build-isolation
|
19 |
-
gradio
|
|
|
1 |
+
torch>=2.0.0
|
2 |
+
transformers>=4.36.0
|
3 |
+
peft>=0.7.0
|
4 |
+
accelerate>=0.25.0
|
5 |
+
bitsandbytes>=0.41.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|