20 lines
840 B
Docker
20 lines
840 B
Docker
|
# dockerfiles/Dockerfile.hermes-local-image
|
||
|
|
FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-runtime@sha256:77f17f843507062875ce8be2a6f76aa6aa3df7f9ef1e31d9d7432f4b0f563dee
|
||
|
|
|
||
|
|
ARG DIFFUSERS_COMMIT=90c0ffdc045902a3667d473d2fbfc03e8716dba9
|
||
|
|
RUN python -m pip install --no-cache-dir \
|
||
|
|
"https://github.com/huggingface/diffusers/archive/${DIFFUSERS_COMMIT}.tar.gz" \
|
||
|
|
"transformers==5.15.0" \
|
||
|
|
"accelerate==1.14.0" \
|
||
|
|
"safetensors==0.8.0" \
|
||
|
|
"huggingface-hub==1.27.0" \
|
||
|
|
"Pillow==12.3.0"
|
||
|
|
RUN python -c "from diffusers import Flux2KleinPipeline; print(Flux2KleinPipeline.__name__)"
|
||
|
|
|
||
|
|
COPY dockerfiles/hermes-local-image-server.py /opt/hermes/local_image_server.py
|
||
|
|
ENV PYTHONUNBUFFERED=1 \
|
||
|
|
HF_HOME=/models/huggingface \
|
||
|
|
HF_HUB_DISABLE_TELEMETRY=1
|
||
|
|
EXPOSE 9004
|
||
|
|
ENTRYPOINT ["python", "/opt/hermes/local_image_server.py"]
|