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