accelerate
diffusers
invisible_watermark
torch==2.1.0
transformers
xformers
fairscale
tensorboard
gradio
torchdiffeq
click
torchvision==0.16.0
pytorch-cuda==12.1
flash_attn==2.5.9.post1