| ARG RELEASE |
| ARG LAUNCHPAD_BUILD_ARCH |
| LABEL org.opencontainers.image.ref.name=ubuntu |
| LABEL org.opencontainers.image.version=22.04 |
| ADD file:415bbc01dfb447d002e2d8173e113ef025d2bbfa20f1205823fa699dc87a2019 in / |
| CMD ["/bin/bash"] |
| ARG PYTORCH_VERSION=2.8.0 |
| ARG TRITON_VERSION= |
| ARG TARGETPLATFORM=linux/amd64 |
| ARG CUDA_VERSION=12.6.3 |
| LABEL com.nvidia.volumes.needed=nvidia_driver |
| RUN |4 PYTORCH_VERSION=2.8.0 TRITON_VERSION= TARGETPLATFORM=linux/amd64 CUDA_VERSION=12.6.3 /bin/sh -c apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends ca-certificates libjpeg-dev libpng-dev && rm -rf /var/lib/apt/lists/* # buildkit |
| COPY /opt/conda /opt/conda # buildkit |
| RUN |4 PYTORCH_VERSION=2.8.0 TRITON_VERSION= TARGETPLATFORM=linux/amd64 CUDA_VERSION=12.6.3 /bin/sh -c if test -n "${TRITON_VERSION}" -a "${TARGETPLATFORM}" != "linux/arm64"; then DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends gcc; rm -rf /var/lib/apt/lists/*; fi # buildkit |
| ENV PATH=/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV NVIDIA_VISIBLE_DEVICES=all |
| ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility |
| ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64 |
| ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV PYTORCH_VERSION=2.8.0 |
| WORKDIR /workspace |
| ENV DEBIAN_FRONTEND=noninteractive PIP_NO_CACHE_DIR=1 HF_HUB_DISABLE_TELEMETRY=1 |
| RUN /bin/sh -c apt-get update && apt-get install -y --no-install-recommends git && rm -rf /var/lib/apt/lists/* # buildkit |
| RUN /bin/sh -c pip install --no-cache-dir "git+https://github.com/huggingface/diffusers" "transformers>=4.57.0" accelerate safetensors sentencepiece protobuf fastapi "uvicorn[standard]" pydantic # buildkit |
| COPY flux_server.py /app/flux_server.py # buildkit |
| WORKDIR /app |
| ENV FLUX_MODEL_DIR=/models/flux2-klein-4b FLUX_HOST=127.0.0.1 FLUX_PORT=8095 FLUX_STEPS=4 |
| CMD ["python" "flux_server.py"] |
