sdeepanraj commited on
Commit
f42d719
·
verified ·
1 Parent(s): 05b5169

Update requirements.txt

Browse files
Files changed (1) hide show
  1. 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