accelerate
diffusers
invisible_watermark
torch
transformers
xformers
torchsde
timm