File size: 224 Bytes
2f9551b
c59ec53
 
 
 
 
2f9551b
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

trl>=0.4.7
flash_attn==2.7.4.post1
transformers==4.48.2
accelerate==1.3.0
peft==0.13.2
torch==1.13.1+cu117  # CUDA version might vary depending on your system
datasets==2.10.1
trfl==0.5.6
tqdm==4.66.0
tokenizers==0.13.2