diff --git a/environment.yml b/environment.yml index a26fc1c0fd..bb44fbe5b3 100644 --- a/environment.yml +++ b/environment.yml @@ -1,14 +1,15 @@ name: invokeai channels: - pytorch - - nvidia - defaults dependencies: - python>=3.9 - pip=22.2.2 - numpy=1.23.3 - - torchvision=0.14.0 - - pytorch-cuda=11.7 + - torchvision=0.13.1 + - torchaudio=0.12.1 + - pytorch=1.12.1 + - cudatoolkit=11.3 - pip: - albumentations==0.4.3 - opencv-python==4.5.5.64 @@ -16,8 +17,7 @@ dependencies: - imageio==2.9.0 - imageio-ffmpeg==0.4.2 - pytorch-lightning==1.7.7 - - omegaconf==2.1.1 - - realesrgan==0.2.5.0 + - omegaconf==2.2.3 - test-tube>=0.7.5 - streamlit==1.12.0 - send2trash==1.8.0 @@ -38,6 +38,7 @@ dependencies: - -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/invoke-ai/Real-ESRGAN.git#egg=realesrgan - -e git+https://github.com/TencentARC/GFPGAN.git#egg=gfpgan - -e git+https://github.com/invoke-ai/clipseg.git@models-rename#egg=clipseg - -e .