transformers
torch
diffusers
accelerate==v0.17.0