LLM/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-02-01 16:35:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' into wan-layerwise-upcasting-tests

Browse Source
This commit is contained in:
Sayak Paul
2026-01-28 17:24:40 +05:30
committed by GitHub
parent b1f936f191 2c669e8480
commit cbfc6ea163
2 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
docker/diffusers-pytorch-cuda/Dockerfile
View File

@@ -1,8 +1,8 @@
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"
ARG PYTHON_VERSION=3.11
ARG PYTHON_VERSION=3.10
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get -y update \
@@ -36,8 +36,12 @@ 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
# 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]"

12
docker/diffusers-pytorch-xformers-cuda/Dockerfile
View File

@@ -1,8 +1,8 @@
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"
ARG PYTHON_VERSION=3.11
ARG PYTHON_VERSION=3.10
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get -y update \
@@ -36,8 +36,12 @@ 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
# 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]"