Update requirements.txt
Browse files- requirements.txt +9 -1
requirements.txt
CHANGED
|
@@ -1,7 +1,15 @@
|
|
|
|
|
| 1 |
streamlit>=1.32.0
|
| 2 |
torch>=2.1.0
|
| 3 |
transformers>=4.36.2
|
| 4 |
peft>=0.6.0
|
| 5 |
pillow>=10.0.0
|
| 6 |
bitsandbytes>=0.42.0
|
| 7 |
-
accelerate>=0.25.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Core dependencies
|
| 2 |
streamlit>=1.32.0
|
| 3 |
torch>=2.1.0
|
| 4 |
transformers>=4.36.2
|
| 5 |
peft>=0.6.0
|
| 6 |
pillow>=10.0.0
|
| 7 |
bitsandbytes>=0.42.0
|
| 8 |
+
accelerate>=0.25.0
|
| 9 |
+
|
| 10 |
+
# Optional but recommended for better performance
|
| 11 |
+
unsloth>=0.4.0
|
| 12 |
+
flash-attn>=2.0.0
|
| 13 |
+
|
| 14 |
+
# For efficient memory handling
|
| 15 |
+
safetensors>=0.4.0
|