accelerate
diffusers
invisible_watermark
torch
transformers
torchvision
timm
loguru
einops
sentencepiece