torch==2.3.0
diffusers==0.28.2
einops==0.6.1
transformers==4.44.2
sentencepiece==0.2.0
accelerate ==0.34.2