scikit-learn==1.2.2
diffusers==0.21.4
transformers==4.34.0
accelerate==0.23.0
einops