requirements.txt

# accelerate==1.4.0
diffusers==0.32.2
torch==2.7.0
torchvision>=0.20.0
black==24.2.0
einops==0.7.0
fire>=0.5.0
gradio==5.29.0
isort==5.13.2
lightning==2.5.0
lpips==0.1.4
opencv-python==4.9.0.80
peft==0.9.0
pydantic>=2.6.1
scipy>=1.12.0
sentencepiece>=0.2.0
timm==0.9.16
tokenizers>=0.15.2
torch-fidelity>=0.3.0
torchmetrics>=1.3.1
transformers==4.42.3
wandb==0.16.2
webdataset>=0.2.86
kornia==0.8.0