Pillow einops safetensors timm tomesd torch torchdiffeq torchsde decord datasets numpy scikit-image opencv-python omegaconf SentencePiece albumentations imageio[ffmpeg] imageio[pyav] tensorboard beautifulsoup4 ftfy func_timeout huggingface_hub accelerate>=0.26.0 diffusers>=0.28.2 transformers>=4.37.2