tqdm
einops
pytorch_lightning
accelerate>=0.20.0
torchsde
pycocotools
diffusers== 0.32.2
timm
transformers==4.49
torch>=2.0.0
opencv-python
omegaconf
gradio==3.45.0
peft==0.17.1