deasdutta commited on
Commit
2148df8
·
verified ·
1 Parent(s): c53971a

Upload requirements.txt with huggingface_hub

Browse files
Files changed (1) hide show
  1. requirements.txt +20 -0
requirements.txt ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Base dependencies
2
+ torch>=2.0.0
3
+ transformers>=4.34.0
4
+ huggingface_hub>=0.17.1
5
+ peft>=0.5.0
6
+ trl>=0.7.4
7
+ datasets>=2.14.0
8
+ llama-cpp-python>=0.2.0
9
+ fastapi>=0.103.2
10
+ uvicorn>=0.23.2
11
+ pydantic>=2.4.2
12
+ numpy>=1.24.0
13
+
14
+ # Training dependencies
15
+ accelerate>=0.23.0
16
+ bitsandbytes>=0.41.0
17
+
18
+ # Utils
19
+ requests>=2.31.0
20
+ tqdm>=4.66.1