torch
diffusers
numpy
pillow
transformers
mediapy
ipywidgets
einops
accelerate