Spaces:
Sleeping
Sleeping
Update requirements.txt
Browse files- requirements.txt +6 -7
requirements.txt
CHANGED
@@ -1,10 +1,9 @@
|
|
1 |
torch>=2.1.0 --index-url https://download.pytorch.org/whl/cpu
|
2 |
-
transformers>=4.40.0
|
3 |
accelerate>=0.25.0
|
4 |
-
# bitsandbytes #
|
5 |
peft>=0.10.0
|
6 |
-
trl>=0.8.0
|
7 |
-
gradio>=4.
|
8 |
-
|
9 |
-
protobuf
|
10 |
-
numpy==1.26.4
|
|
|
1 |
torch>=2.1.0 --index-url https://download.pytorch.org/whl/cpu
|
2 |
+
transformers[sentencepiece]>=4.40.0 # Added sentencepiece here
|
3 |
accelerate>=0.25.0
|
4 |
+
# bitsandbytes # Optional for CPU, can remove if causing issues
|
5 |
peft>=0.10.0
|
6 |
+
# trl>=0.8.0 # Not strictly needed for inference app, only for training
|
7 |
+
gradio>=4.15.0 # Try a version known to be stable or newer if buttons are desired
|
8 |
+
numpy==1.26.4
|
9 |
+
protobuf # Often a dependency for tokenizers
|
|