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