torch
transformers
IPython
diffusers
accelerate