accelerate==0.14.0
transformers>=4.21.0
ftfy
albumentations
opencv-python
einops
diffusers[torch]==0.9.0
pytorch_lightning
bitsandbytes==0.35.0
tensorboard