torch
ml-collections
diffusers==0.27.2
transformers
accelerate