FROM runpod/stable-diffusion:web-automatic-base-4.0.0 AS build

SHELL ["/bin/bash", "-c"]

ENV PATH="${PATH}:/workspace/stable-diffusion-webui/venv/bin"
ENV PYTHONUNBUFFERED=1

RUN pip install -U xformers

WORKDIR /

COPY model.ckpt /workspace/stable-diffusion-webui/models/Stable-diffusion/model.ckpt
COPY new15.safetensors /workspace/stable-diffusion-webui/models/Lora/new15.safetensors

FROM build AS with_logo

ADD custom-start.sh /workspace/stable-diffusion-webui/custom-start.sh
RUN chmod +x /workspace/stable-diffusion-webui/custom-start.sh

WORKDIR /workspace/stable-diffusion-webui

CMD [ "sleep", "infinity" ]