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 diffusers>=0.30.1 transformers==4.47 av==12.0.0 gradio