diff --git a/requirements-lin-win-colab-CUDA.txt b/requirements-lin-win-colab-CUDA.txt index 91c8d0f35c..bb6a9ae9c8 100644 --- a/requirements-lin-win-colab-CUDA.txt +++ b/requirements-lin-win-colab-CUDA.txt @@ -2,6 +2,6 @@ # Get hardware-appropriate torch/torchvision --extra-index-url https://download.pytorch.org/whl/cu116 --trusted-host https://download.pytorch.org -torch -torchvision +torch==1.12.1 +torchvision==0.13.1 -e . diff --git a/requirements.txt b/requirements.txt index 1618e4a427..939463e36e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -36,5 +36,6 @@ realesrgan diffusers git+https://github.com/openai/CLIP.git@main#egg=clip git+https://github.com/Birch-san/k-diffusion.git@mps#egg=k-diffusion -git+https://github.com/TencentARC/GFPGAN.git#egg=gfpgan +git+https://github.com/invoke-ai/Real-ESRGAN.git#egg=realesrgan +git+https://github.com/invoke-ai/GFPGAN.git#egg=gfpgan -e git+https://github.com/invoke-ai/clipseg.git@models-rename#egg=clipseg