diff --git a/environment-mac.yaml b/environment-mac.yaml index 8338d632384..1466cc3193a 100644 --- a/environment-mac.yaml +++ b/environment-mac.yaml @@ -32,7 +32,7 @@ dependencies: - omegaconf==2.1.1 - onnx==1.12.0 - onnxruntime==1.12.1 - - opencv==4.6.0 + - opencv-python==4.5.5.64 - pudb==2022.1 - pytorch-lightning==1.6.5 - realesrgan==0.2.5.0 diff --git a/environment.yaml b/environment.yaml index 2a8a6290bed..db754a91858 100644 --- a/environment.yaml +++ b/environment.yaml @@ -17,6 +17,7 @@ dependencies: - imageio-ffmpeg==0.4.2 - pytorch-lightning==1.4.2 - omegaconf==2.1.1 + - realesrgan==0.2.5.0 - test-tube>=0.7.5 - streamlit==1.12.0 - pillow==9.2.0 diff --git a/requirements-lin.txt b/requirements-lin.txt index 9c3d1e7451c..3f22122eec1 100644 --- a/requirements-lin.txt +++ b/requirements-lin.txt @@ -24,6 +24,7 @@ torchmetrics==0.6.0 transformers==4.19.2 git+https://github.com/openai/CLIP.git@main#egg=clip git+https://github.com/lstein/k-diffusion.git@master#egg=k-diffusion +git+https://github.com/lstein/GFPGAN@fix-dark-cast-images#egg=gfpgan # No CUDA in PyPi builds --extra-index-url https://download.pytorch.org/whl/cu113 --trusted-host https://download.pytorch.org torch==1.11.0 diff --git a/requirements-mac.txt b/requirements-mac.txt index 8fc4f4fa7ea..4cbdf04cd51 100644 --- a/requirements-mac.txt +++ b/requirements-mac.txt @@ -21,3 +21,4 @@ transformers==4.19.2 -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/Birch-san/k-diffusion.git@mps#egg=k-diffusion +- -e git+https://github.com/lstein/GFPGAN@fix-dark-cast-images#egg=gfpgan diff --git a/requirements-win.txt b/requirements-win.txt index 9c3d1e7451c..3f22122eec1 100644 --- a/requirements-win.txt +++ b/requirements-win.txt @@ -24,6 +24,7 @@ torchmetrics==0.6.0 transformers==4.19.2 git+https://github.com/openai/CLIP.git@main#egg=clip git+https://github.com/lstein/k-diffusion.git@master#egg=k-diffusion +git+https://github.com/lstein/GFPGAN@fix-dark-cast-images#egg=gfpgan # No CUDA in PyPi builds --extra-index-url https://download.pytorch.org/whl/cu113 --trusted-host https://download.pytorch.org torch==1.11.0