requirements.txt

# torch==2.1.1
# torchaudio==2.1.1
# torchvision==0.16.1
# diffusers
mmcv==1.7.0
# git+https://github.com/huggingface/diffusers
# timm==0.6.12
accelerate
tensorboard
tensorboardX
transformers
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==0.6.2