From 284228fdd11d6bbd96e20b42069d2370bd341712 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Thu, 11 May 2023 19:22:25 +0000 Subject: [PATCH 1/2] up --- docker/diffusers-pytorch-cuda/Dockerfile | 1 + 1 file changed, 1 insertion(+) diff --git a/docker/diffusers-pytorch-cuda/Dockerfile b/docker/diffusers-pytorch-cuda/Dockerfile index 6946685be280..6a8afb54df7e 100644 --- a/docker/diffusers-pytorch-cuda/Dockerfile +++ b/docker/diffusers-pytorch-cuda/Dockerfile @@ -27,6 +27,7 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip && \ torch \ torchvision \ torchaudio \ + --extra-index-url https://download.pytorch.org/whl/cu118 && \ python3 -m pip install --no-cache-dir \ accelerate \ datasets \ From d978b95051722b99dbfa460b053e96c7b9ab28ef Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Thu, 11 May 2023 19:27:49 +0000 Subject: [PATCH 2/2] up --- docker/diffusers-pytorch-cuda/Dockerfile | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/docker/diffusers-pytorch-cuda/Dockerfile b/docker/diffusers-pytorch-cuda/Dockerfile index 6a8afb54df7e..a51a12ee2838 100644 --- a/docker/diffusers-pytorch-cuda/Dockerfile +++ b/docker/diffusers-pytorch-cuda/Dockerfile @@ -26,8 +26,7 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip && \ python3 -m pip install --no-cache-dir \ torch \ torchvision \ - torchaudio \ - --extra-index-url https://download.pytorch.org/whl/cu118 && \ + torchaudio && \ python3 -m pip install --no-cache-dir \ accelerate \ datasets \