scikit-learn==1.2.2
diffusers==0.25.1
transformers==4.37.2
accelerate==0.26.1
einops