requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu117
torch==1.13.1
torchvision==0.14.1
diffusers==0.12.1
transformers==4.26.0
numpy==1.24.2
seaborn==0.12.2
accelerate==0.16.0
scikit-learn==0.24.1