torch
safetensors
diffusers
transformers
einops
toml
