torch==2.3.0
diffusers==0.28.2
transformers==4.44.2
sentencepiece
accelerate
einops