requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu113
torch==1.11.0
torchvision==0.12.0
diffusers==0.12.1
transformers==4.25.1
numpy==1.24.2
seaborn==0.12.2
accelerate==0.18.0
scikit-learn==0.24.1