215dd59 931641c 215dd59 df076e1 4b86f56 215dd59 271be34 215dd59
1
2
3
4
5
6
7
8
9
10
11
12
13
accelerate==0.30.1 lpips==0.1.4 numpy<2 matplotlib transformers==4.41.0 torch==2.2.2 torchvision lightning>2.3.1 tqdm==4.66.4 wandb xformers==0.0.25.post1 git+https://github.com/janEbert/mup.git@fsdp-fix