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