Spaces:
Sleeping
Sleeping
Update requirements.txt
Browse files- requirements.txt +6 -2
requirements.txt
CHANGED
@@ -1,8 +1,12 @@
|
|
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
|
7 |
sentencepiece # tokenizer dependency
|
8 |
-
gradio>=4.28
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
transformers>=4.46.0
|
2 |
optimum>=1.23.0 # makes transformers’ quantizer happy
|
3 |
auto-gptq>=0.7.0.post0 # wheels available for CUDA 11.8/12.x
|
4 |
bitsandbytes>=0.43.0 # still useful for 4-bit mat-muls
|
5 |
accelerate>=0.29.0
|
6 |
sentencepiece # tokenizer dependency
|
7 |
+
gradio>=4.28
|
8 |
+
# requirements.txt – CPU-only
|
9 |
+
transformers>=4.46.0 # model & pipeline
|
10 |
+
torch>=2.2.0 # CPU wheel auto-selected
|
11 |
+
sentencepiece>=0.1.99 # DistilBART tokenizer
|
12 |
+
gradio>=4.28.0 # UI
|