215dd59
931641c
215dd59
df076e1
4b86f56
215dd59
271be34
215dd59 | | 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
|
