# x86_64 CUDA 13 image for RunPod Blackwell distributed validation.
FROM pytorch/pytorch:2.9.1-cuda13.0-cudnn9-devel

ARG COMFY_KITCHEN_VERSION=0.2.31

RUN apt-get update \
    && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends ffmpeg git \
    && rm -rf /var/lib/apt/lists/*

WORKDIR /opt/h3-blackwell-runtime
COPY . .

RUN python -m pip install --no-cache-dir \
    "comfy-kitchen==${COMFY_KITCHEN_VERSION}" \
    "fastsafetensors>=0.1.10" \
    "numpy>=2.0" \
    "pillow>=11" \
    "safetensors>=0.5.0" \
    "scipy>=1.14" \
    "transformers>=4.51,<5"

ENV PYTHONPATH=/opt/h3-blackwell-runtime/src
ENTRYPOINT []
CMD ["bash"]