mirror of
https://github.com/huggingface/diffusers.git
synced 2026-10-24 20:59:27 +03:00
efe46a5b17
* move to python 3.11. * update setup.py along with others. * fix(ci): pin GitHub Actions to commit SHAs (#15030) (#15033) fix(ci): pin actions in workflows flagged on #15030 Co-authored-by: hf-security-analysis[bot] <265538906+hf-security-analysis[bot]@users.noreply.github.com> * up * fix(ci): pin GitHub Actions to commit SHAs (#15030) (#15036) fix(ci): pin actions in workflows flagged on #15030 Co-authored-by: hf-security-analysis[bot] <265538906+hf-security-analysis[bot]@users.noreply.github.com> * up --------- Co-authored-by: hf-security-analysis[bot] <265538906+hf-security-analysis[bot]@users.noreply.github.com>
46 lines
1.1 KiB
Docker
46 lines
1.1 KiB
Docker
FROM python:3.11-slim
|
|
ENV PYTHONDONTWRITEBYTECODE=1
|
|
LABEL maintainer="Hugging Face"
|
|
LABEL repository="diffusers"
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
|
RUN apt-get -y update && apt-get install -y bash \
|
|
build-essential \
|
|
git \
|
|
git-lfs \
|
|
curl \
|
|
ca-certificates \
|
|
libglib2.0-0 \
|
|
libsndfile1-dev \
|
|
libgl1 \
|
|
zip \
|
|
wget
|
|
|
|
ENV UV_PYTHON=/usr/local/bin/python
|
|
|
|
# pre-install the heavy dependencies (these can later be overridden by the deps from setup.py)
|
|
RUN pip install uv
|
|
RUN uv pip install --no-cache-dir \
|
|
torch==2.10.0 \
|
|
torchvision==0.25.0 \
|
|
torchaudio==2.10.0 \
|
|
--extra-index-url https://download.pytorch.org/whl/cpu
|
|
|
|
RUN uv pip install --no-cache-dir "git+https://github.com/huggingface/diffusers.git@main#egg=diffusers[test]"
|
|
|
|
# Extra dependencies
|
|
RUN uv pip install --no-cache-dir \
|
|
accelerate \
|
|
numpy==1.26.4 \
|
|
hf_xet \
|
|
setuptools==69.5.1 \
|
|
bitsandbytes \
|
|
torchao \
|
|
gguf \
|
|
optimum-quanto
|
|
|
|
RUN apt-get clean && rm -rf /var/lib/apt/lists/* && apt-get autoremove && apt-get autoclean
|
|
|
|
CMD ["/bin/bash"]
|