pytorch_lightning==2.1.2
deepspeed==0.15.0
taming-transformers==0.0.1
diffusers==0.30.1
transformers
torchvision==0.16.2
scipy==1.14.1
einops==0.8.0
kornia==0.7.3
open_clip_torch==2.20.0
openai-clip==1.0.1
numpy==1.26.3
xformers==0.0.23.post1
timm==1.0.9
av==12.3.0
gradio==5.8.0