# 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"]