File size: 169 Bytes
4b66da1
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
# Requirements for model training - pinned versions for compatibility
torch
transformers==4.44.2
datasets
peft
bitsandbytes
trl==0.8.6
accelerate
scipy
hf_transfer
rich