--index-url https://download.pytorch.org/whl/cu124
--extra-index-url https://pypi.org/simple
torch==2.4.1+cu124
torchaudio==2.4.1+cu124
torchvision==0.19.1+cu124
albumentations==1.4.15
pytorch-lightning==2.4.0
opencv-python==4.10.0.84
imageio==2.35.1
omegaconf==2.3.0
einops==0.8.0
torch-fidelity==0.3.0
transformers==4.44.2
wandb==0.18.1
kornia==0.7.3
scikit-learn==1.5.2
pandas==2.2.3
pytorch-fid==0.3.0
-e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
-e git+https://github.com/openai/CLIP.git@main#egg=clip
-e ./latent-diffusion