FROM nvidia/cuda:11.6.1-runtime-ubuntu20.04

RUN apt update && apt -y install python3-pip python-is-python3 && \
  pip3 install -U pip && \
  rm -rf /var/lib/apt/lists/* /var/log* /tmp/* /var/tmp/*

WORKDIR /app
COPY ./requirements.txt ./
RUN pip install --no-cache -r requirements.txt

COPY . ./

#ENV SAVE_FILES="1"

ENV DTYPE="float32"
ENV CLIP_CONTEXTS=3
ENV AMP="1"
ENV MODEL="stable-diffusion"
ENV DEV="True"
ENV MODEL_PATH="models/animefull-final-pruned"
#these aren't actually used by the site
ENV MODULE_PATH="models/modules"
#unclear if these are used either
ENV PRIOR_PATH="models/vector_adjust_v2.pt"
ENV ENABLE_EMA="1"
ENV VAE_PATH="models/animevae.pt"
ENV PENULTIMATE="1"
ENV PYTHONDONTWRITEBYTECODE=1
ENV VRAM_MODE="medvram"
 
 CMD ["python3", "-m", "uvicorn", "--host", "0.0.0.0", "--port=6969", "main:app"]
