Pillow einops safetensors timm tomesd torchdiffeq torchsde xformers decord datasets numpy scikit-image opencv-python omegaconf SentencePiece albumentations imageio[ffmpeg] imageio[pyav] tensorboard beautifulsoup4 ftfy func_timeout deepspeed accelerate>=0.25.0,<1.0 diffusers==0.32.1 transformers>=4.40,<4.47 av==12.0.0 gradio