Pillow einops safetensors timm tomesd accelerate torch>=2.2.0 torchdiffeq torchsde xformers decord datasets numpy scikit-image opencv-python omegaconf SentencePiece albumentations imageio[ffmpeg] imageio[pyav] tensorboard gradio==3.41.2 diffusers==0.27.0 transformers==4.37.2