From 215e6804d3bb011faead6ddc8db00cdbe31b6e91 Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Mon, 12 Feb 2024 16:01:05 +0530 Subject: [PATCH] Unpin torch versions in CI (#6945) * update * update * update --------- Co-authored-by: Sayak Paul --- docker/diffusers-pytorch-compile-cuda/Dockerfile | 6 +++--- docker/diffusers-pytorch-cpu/Dockerfile | 6 +++--- docker/diffusers-pytorch-cuda/Dockerfile | 6 +++--- docker/diffusers-pytorch-xformers-cuda/Dockerfile | 6 +++--- setup.py | 4 ++-- src/diffusers/dependency_versions_table.py | 4 ++-- 6 files changed, 16 insertions(+), 16 deletions(-) diff --git a/docker/diffusers-pytorch-compile-cuda/Dockerfile b/docker/diffusers-pytorch-compile-cuda/Dockerfile index 5e6be6831e..4b3841a576 100644 --- a/docker/diffusers-pytorch-compile-cuda/Dockerfile +++ b/docker/diffusers-pytorch-compile-cuda/Dockerfile @@ -26,9 +26,9 @@ ENV PATH="/opt/venv/bin:$PATH" # pre-install the heavy dependencies (these can later be overridden by the deps from setup.py) RUN python3.9 -m pip install --no-cache-dir --upgrade pip && \ python3.9 -m pip install --no-cache-dir \ - torch==2.1.2 \ - torchvision==0.16.2 \ - torchaudio==2.1.2 \ + torch \ + torchvision \ + torchaudio \ invisible_watermark && \ python3.9 -m pip install --no-cache-dir \ accelerate \ diff --git a/docker/diffusers-pytorch-cpu/Dockerfile b/docker/diffusers-pytorch-cpu/Dockerfile index 66412d7502..127c61a719 100644 --- a/docker/diffusers-pytorch-cpu/Dockerfile +++ b/docker/diffusers-pytorch-cpu/Dockerfile @@ -25,9 +25,9 @@ ENV PATH="/opt/venv/bin:$PATH" # pre-install the heavy dependencies (these can later be overridden by the deps from setup.py) RUN python3 -m pip install --no-cache-dir --upgrade pip && \ python3 -m pip install --no-cache-dir \ - torch==2.1.2 \ - torchvision==0.16.2 \ - torchaudio==2.1.2 \ + torch \ + torchvision \ + torchaudio \ invisible_watermark \ --extra-index-url https://download.pytorch.org/whl/cpu && \ python3 -m pip install --no-cache-dir \ diff --git a/docker/diffusers-pytorch-cuda/Dockerfile b/docker/diffusers-pytorch-cuda/Dockerfile index efc995e26b..7fc3d8ced9 100644 --- a/docker/diffusers-pytorch-cuda/Dockerfile +++ b/docker/diffusers-pytorch-cuda/Dockerfile @@ -25,9 +25,9 @@ ENV PATH="/opt/venv/bin:$PATH" # pre-install the heavy dependencies (these can later be overridden by the deps from setup.py) RUN python3 -m pip install --no-cache-dir --upgrade pip && \ python3 -m pip install --no-cache-dir \ - torch==2.1.2 \ - torchvision==0.16.2 \ - torchaudio==2.1.2 \ + torch \ + torchvision \ + torchaudio \ invisible_watermark && \ python3 -m pip install --no-cache-dir \ accelerate \ diff --git a/docker/diffusers-pytorch-xformers-cuda/Dockerfile b/docker/diffusers-pytorch-xformers-cuda/Dockerfile index 56a46ab865..8f2619c623 100644 --- a/docker/diffusers-pytorch-xformers-cuda/Dockerfile +++ b/docker/diffusers-pytorch-xformers-cuda/Dockerfile @@ -25,9 +25,9 @@ ENV PATH="/opt/venv/bin:$PATH" # pre-install the heavy dependencies (these can later be overridden by the deps from setup.py) RUN python3 -m pip install --no-cache-dir --upgrade pip && \ python3 -m pip install --no-cache-dir \ - torch==2.1.2 \ - torchvision==0.16.2 \ - torchaudio==2.1.2 \ + torch \ + torchvision \ + torchaudio \ invisible_watermark && \ python3 -m pip install --no-cache-dir \ accelerate \ diff --git a/setup.py b/setup.py index 6773bc1642..b97b01fb4a 100644 --- a/setup.py +++ b/setup.py @@ -126,8 +126,8 @@ _deps = [ "regex!=2019.12.17", "requests", "tensorboard", - "torch>=1.4,<2.2.0", - "torchvision<0.17", + "torch>=1.4", + "torchvision", "transformers>=4.25.1", "urllib3<=2.0.0", ] diff --git a/src/diffusers/dependency_versions_table.py b/src/diffusers/dependency_versions_table.py index 618cac807c..e92a486bff 100644 --- a/src/diffusers/dependency_versions_table.py +++ b/src/diffusers/dependency_versions_table.py @@ -38,8 +38,8 @@ deps = { "regex": "regex!=2019.12.17", "requests": "requests", "tensorboard": "tensorboard", - "torch": "torch>=1.4,<2.2.0", - "torchvision": "torchvision<0.17", + "torch": "torch>=1.4", + "torchvision": "torchvision", "transformers": "transformers>=4.25.1", "urllib3": "urllib3<=2.0.0", }