numpy
accelerate
diffusers
invisible_watermark
torch
transformers==4.38.2
xformers
sentencepiece
datasets
einops
Pillow
torchvision
tqdm
imwatermark