numpy==1.26.4
torch==2.1.1
torchaudio==2.1.1
torchvision==0.16.1
accelerate
diffusers
invisible_watermark
transformers
timm==0.6.12
tensorboard
tensorboardX
sentencepiece~=0.1.99
ftfy
beautifulsoup4
protobuf==3.20.2
gradio==4.1.1
yapf==0.40.1
opencv-python
bs4
einops
xformers
optimum
peft