File size: 204 Bytes
215dd59
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
accelerate==0.30.1
torchvision==0.18.0
lpips==0.1.4
matplotlib
transformers==4.41.0
torch==2.3.0
lightning>2.3.1
tqdm==4.66.4
wandb
xformers==0.0.26.post1
git+https://github.com/janEbert/mup.git@fsdp-fix