diff --git a/requirements.txt b/requirements.txt index 9637d43..090b52d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,7 +4,7 @@ opencv-python==4.1.2.30 pudb==2019.2 imageio==2.9.0 imageio-ffmpeg==0.4.2 -pytorch-lightning==1.4.2 +pytorch-lightning==1.6.5 omegaconf==2.1.1 test-tube>=0.7.5 streamlit>=0.73.1 @@ -14,6 +14,6 @@ transformers==4.19.2 torchmetrics==0.6.0 kornia==0.6 gradio -git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers +git+https://github.com/illeatmyhat/taming-transformers.git@master#egg=taming-transformers git+https://github.com/openai/CLIP.git@main#egg=clip git+https://github.com/hlky/k-diffusion-sd#egg=k_diffusion