diff --git a/environments-and-requirements/environment-win-cuda.yml b/environments-and-requirements/environment-win-cuda.yml index a75711cfe8..b2a919882b 100644 --- a/environments-and-requirements/environment-win-cuda.yml +++ b/environments-and-requirements/environment-win-cuda.yml @@ -13,6 +13,7 @@ dependencies: - cudatoolkit=11.6 - pip: - albumentations==0.4.3 + - basicsr==1.4.1 - dependency_injector==4.40.0 - diffusers==0.6.0 - einops==0.3.0 diff --git a/environments-and-requirements/requirements-win-colab-cuda.txt b/environments-and-requirements/requirements-win-colab-cuda.txt index 68ee1bace6..b00f1487fb 100644 --- a/environments-and-requirements/requirements-win-colab-cuda.txt +++ b/environments-and-requirements/requirements-win-colab-cuda.txt @@ -2,6 +2,7 @@ # Get hardware-appropriate torch/torchvision --extra-index-url https://download.pytorch.org/whl/cu116 --trusted-host https://download.pytorch.org +basicsr==1.4.1 torch==1.12.1 torchvision==0.13.1 -e .