File size: 327 Bytes
2148df8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# Base dependencies
torch>=2.0.0
transformers>=4.34.0
huggingface_hub>=0.17.1
peft>=0.5.0
trl>=0.7.4
datasets>=2.14.0
llama-cpp-python>=0.2.0
fastapi>=0.103.2
uvicorn>=0.23.2
pydantic>=2.4.2
numpy>=1.24.0

# Training dependencies
accelerate>=0.23.0
bitsandbytes>=0.41.0

# Utils
requests>=2.31.0
tqdm>=4.66.1