From 58604dd1b9c31cebeab20d78b9c3b23e67e1d9da Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Wed, 28 Jan 2026 16:33:06 +0530 Subject: [PATCH 1/4] change to CUDA 12.9. --- docker/diffusers-pytorch-cuda/Dockerfile | 5 ++--- docker/diffusers-pytorch-xformers-cuda/Dockerfile | 5 ++--- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/docker/diffusers-pytorch-cuda/Dockerfile b/docker/diffusers-pytorch-cuda/Dockerfile index 134b47215d76..26e661d93d6b 100644 --- a/docker/diffusers-pytorch-cuda/Dockerfile +++ b/docker/diffusers-pytorch-cuda/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:12.1.0-runtime-ubuntu20.04 +FROM nvidia/cuda:12.9.0-runtime-ubuntu20.04 LABEL maintainer="Hugging Face" LABEL repository="diffusers" @@ -36,8 +36,7 @@ ENV PATH="$VIRTUAL_ENV/bin:$PATH" RUN uv pip install --no-cache-dir \ torch \ torchvision \ - torchaudio \ - --index-url https://download.pytorch.org/whl/cu121 + torchaudio RUN uv pip install --no-cache-dir "git+https://github.com/huggingface/diffusers.git@main#egg=diffusers[test]" diff --git a/docker/diffusers-pytorch-xformers-cuda/Dockerfile b/docker/diffusers-pytorch-xformers-cuda/Dockerfile index 7714821c82ad..dbf4b60c2748 100644 --- a/docker/diffusers-pytorch-xformers-cuda/Dockerfile +++ b/docker/diffusers-pytorch-xformers-cuda/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:12.1.0-runtime-ubuntu20.04 +FROM nvidia/cuda:12.9.0-runtime-ubuntu20.04 LABEL maintainer="Hugging Face" LABEL repository="diffusers" @@ -36,8 +36,7 @@ ENV PATH="$VIRTUAL_ENV/bin:$PATH" RUN uv pip install --no-cache-dir \ torch \ torchvision \ - torchaudio \ - --index-url https://download.pytorch.org/whl/cu121 + torchaudio RUN uv pip install --no-cache-dir "git+https://github.com/huggingface/diffusers.git@main#egg=diffusers[test]" From 2a2655ccad6170d319bd621dbb90e1242fda7b3e Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Wed, 28 Jan 2026 16:51:21 +0530 Subject: [PATCH 2/4] up --- docker/diffusers-pytorch-cuda/Dockerfile | 7 ++++++- docker/diffusers-pytorch-xformers-cuda/Dockerfile | 7 ++++++- 2 files changed, 12 insertions(+), 2 deletions(-) diff --git a/docker/diffusers-pytorch-cuda/Dockerfile b/docker/diffusers-pytorch-cuda/Dockerfile index 26e661d93d6b..0c639f853248 100644 --- a/docker/diffusers-pytorch-cuda/Dockerfile +++ b/docker/diffusers-pytorch-cuda/Dockerfile @@ -2,7 +2,7 @@ FROM nvidia/cuda:12.9.0-runtime-ubuntu20.04 LABEL maintainer="Hugging Face" LABEL repository="diffusers" -ARG PYTHON_VERSION=3.11 +ARG PYTHON_VERSION=3.10 ENV DEBIAN_FRONTEND=noninteractive RUN apt-get -y update \ @@ -38,6 +38,11 @@ RUN uv pip install --no-cache-dir \ torchvision \ torchaudio +# Install compatible versions of numba/llvmlite for Python 3.10+ +RUN uv pip install --no-cache-dir \ + "llvmlite>=0.40.0" \ + "numba>=0.57.0" + RUN uv pip install --no-cache-dir "git+https://github.com/huggingface/diffusers.git@main#egg=diffusers[test]" # Extra dependencies diff --git a/docker/diffusers-pytorch-xformers-cuda/Dockerfile b/docker/diffusers-pytorch-xformers-cuda/Dockerfile index dbf4b60c2748..f40df34a4c59 100644 --- a/docker/diffusers-pytorch-xformers-cuda/Dockerfile +++ b/docker/diffusers-pytorch-xformers-cuda/Dockerfile @@ -2,7 +2,7 @@ FROM nvidia/cuda:12.9.0-runtime-ubuntu20.04 LABEL maintainer="Hugging Face" LABEL repository="diffusers" -ARG PYTHON_VERSION=3.11 +ARG PYTHON_VERSION=3.10 ENV DEBIAN_FRONTEND=noninteractive RUN apt-get -y update \ @@ -38,6 +38,11 @@ RUN uv pip install --no-cache-dir \ torchvision \ torchaudio +# Install compatible versions of numba/llvmlite for Python 3.10+ +RUN uv pip install --no-cache-dir \ + "llvmlite>=0.40.0" \ + "numba>=0.57.0" + RUN uv pip install --no-cache-dir "git+https://github.com/huggingface/diffusers.git@main#egg=diffusers[test]" # Extra dependencies From 7e1f61f85fa879dcfad067dba3c2c3b84c8fbf68 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Wed, 28 Jan 2026 16:52:10 +0530 Subject: [PATCH 3/4] change runtime base --- docker/diffusers-pytorch-cuda/Dockerfile | 2 +- docker/diffusers-pytorch-xformers-cuda/Dockerfile | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/diffusers-pytorch-cuda/Dockerfile b/docker/diffusers-pytorch-cuda/Dockerfile index 0c639f853248..6d3c3bae6d0b 100644 --- a/docker/diffusers-pytorch-cuda/Dockerfile +++ b/docker/diffusers-pytorch-cuda/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:12.9.0-runtime-ubuntu20.04 +FROM FROM nvidia/cuda:12.9.0-runtime-ubuntu20.04 LABEL maintainer="Hugging Face" LABEL repository="diffusers" diff --git a/docker/diffusers-pytorch-xformers-cuda/Dockerfile b/docker/diffusers-pytorch-xformers-cuda/Dockerfile index f40df34a4c59..08a1ccabf5a1 100644 --- a/docker/diffusers-pytorch-xformers-cuda/Dockerfile +++ b/docker/diffusers-pytorch-xformers-cuda/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:12.9.0-runtime-ubuntu20.04 +FROM FROM nvidia/cuda:12.9.0-runtime-ubuntu20.04 LABEL maintainer="Hugging Face" LABEL repository="diffusers" From 9872b40e1600a49fb409db39882f812a2a2c1a96 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Wed, 28 Jan 2026 16:54:56 +0530 Subject: [PATCH 4/4] FROM --- docker/diffusers-pytorch-cuda/Dockerfile | 2 +- docker/diffusers-pytorch-xformers-cuda/Dockerfile | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/diffusers-pytorch-cuda/Dockerfile b/docker/diffusers-pytorch-cuda/Dockerfile index 6d3c3bae6d0b..0c639f853248 100644 --- a/docker/diffusers-pytorch-cuda/Dockerfile +++ b/docker/diffusers-pytorch-cuda/Dockerfile @@ -1,4 +1,4 @@ -FROM FROM nvidia/cuda:12.9.0-runtime-ubuntu20.04 +FROM nvidia/cuda:12.9.0-runtime-ubuntu20.04 LABEL maintainer="Hugging Face" LABEL repository="diffusers" diff --git a/docker/diffusers-pytorch-xformers-cuda/Dockerfile b/docker/diffusers-pytorch-xformers-cuda/Dockerfile index 08a1ccabf5a1..f40df34a4c59 100644 --- a/docker/diffusers-pytorch-xformers-cuda/Dockerfile +++ b/docker/diffusers-pytorch-xformers-cuda/Dockerfile @@ -1,4 +1,4 @@ -FROM FROM nvidia/cuda:12.9.0-runtime-ubuntu20.04 +FROM nvidia/cuda:12.9.0-runtime-ubuntu20.04 LABEL maintainer="Hugging Face" LABEL repository="diffusers"