diff --git a/requirements.txt b/requirements.txt index 41530b7306..4c6dcdaab1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,23 +1,23 @@ accelerate==0.12.0 albumentations==0.4.3 +clip==1.0 einops==0.3.0 huggingface-hub==0.8.1 imageio==2.9.0 imageio-ffmpeg==0.4.2 kornia==0.6.0 -numpy=1.19.2 +numpy==1.19.2 omegaconf==2.1.1 opencv-python==4.1.2.30 pudb==2019.2 -pytorch=1.11.0 +pytorch pytorch-lightning==1.4.2 streamlit==1.12.0 test-tube>=0.7.5 torch-fidelity==0.3.0 torchmetrics==0.6.0 -torchvision==0.12.0 +torchvision transformers==4.19.2 --e git+git@github.com:lstein/stable-diffusion.git@cb9e6d544a5ae8f6fe527a86ff3cc3455e9683f3#egg=latent_diffusion --e git+https://github.com/CompVis/taming-transformers.git@24268930bf1dce879235a7fddd0b2355b84d7ea6#egg=taming_transformers --e git+https://github.com/lstein/k-diffusion.git@db5799068749bf3a6d5845120ed32df16b7d883b#egg=k_diffusion --e git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1#egg=clip +-e git+https://github.com/openai/CLIP.git@main#egg=clip +-e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers +-e git+https://github.com/lstein/k-diffusion.git@master#egg=k-diffusion