diffusers==0.3.0
einops==0.7.0
extensions==0.4
lovely_tensors==0.1.15
matplotlib==3.8.3
numpy==1.22.3
scikit_learn==1.4.1.post1
scipy==1.12.0
seaborn==0.13.2
torch==2.1.2+cu118
torch==2.1.2
tqdm==4.66.2
