Spaces:
Sleeping
Sleeping
Update requirements.txt
Browse files- requirements.txt +6 -4
requirements.txt
CHANGED
@@ -1,4 +1,6 @@
|
|
1 |
-
|
2 |
-
|
3 |
-
|
4 |
-
|
|
|
|
|
|
1 |
+
torch>=2.1
|
2 |
+
transformers>=4.46.0
|
3 |
+
optimum>=1.23.0 # makes transformers’ quantizer happy
|
4 |
+
auto-gptq>=0.7.0.post0 # wheels available for CUDA 11.8/12.x
|
5 |
+
bitsandbytes>=0.43.0 # still useful for 4-bit mat-muls
|
6 |
+
accelerate>=0.29.0
|