diffusers
torch
numpy