Update requirements.txt (#32)
Browse files- Update requirements.txt (f42d719c6ba95443a65d9fcb1e8dc2c14b9755c6)
- requirements.txt +10 -2
requirements.txt
CHANGED
@@ -3,7 +3,15 @@ requests
|
|
3 |
smolagents
|
4 |
smolagents[transformers]
|
5 |
duckduckgo-search
|
6 |
-
transformers
|
7 |
wikipedia-api
|
8 |
beautifulsoup4
|
9 |
-
pandas
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
smolagents
|
4 |
smolagents[transformers]
|
5 |
duckduckgo-search
|
|
|
6 |
wikipedia-api
|
7 |
beautifulsoup4
|
8 |
+
pandas
|
9 |
+
# CPU-only PyTorch
|
10 |
+
torch==2.1.2+cpu
|
11 |
+
# Hugging Face Transformers
|
12 |
+
transformers>=4.36.2
|
13 |
+
accelerate>=0.27.2
|
14 |
+
# Tokenizers (sometimes pulled separately)
|
15 |
+
sentencepiece
|
16 |
+
# Optional: still needed for 4-bit loading, even if using GPU
|
17 |
+
bitsandbytes==0.42.0
|