Update Dockerfile

Dockerfile

@@ -3,54 +3,53 @@ FROM nvidia/cuda:12.2.2-cudnn8-devel-ubuntu22.04
 ENV DEBIAN_FRONTEND=noninteractive \
     TZ=America/Los_Angeles
 
+ARG PYTHON_VERSION=3.10.13
 ARG USE_PERSISTENT_DATA
 
 RUN apt-get update && apt-get install -y \
-    git \
+    git git-lfs \
     make build-essential libssl-dev zlib1g-dev \
     libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm \
-    libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev git-lfs  \
+    libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev \
     ffmpeg libsm6 libxext6 cmake libgl1-mesa-glx \
     && rm -rf /var/lib/apt/lists/* \
     && git lfs install
 
-WORKDIR /code
-
-COPY ./requirements.txt /code/requirements.txt
-
-# User
 RUN useradd -m -u 1000 user
 USER user
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH
 
-# Pyenv
 RUN curl https://pyenv.run | bash
 ENV PATH=$HOME/.pyenv/shims:$HOME/.pyenv/bin:$PATH
 
-ARG PYTHON_VERSION=3.10.13
-# Python
 RUN pyenv install $PYTHON_VERSION && \
     pyenv global $PYTHON_VERSION && \
     pyenv rehash && \
-    pip install --no-cache-dir --upgrade pip setuptools wheel && \
-    pip install --no-cache-dir \
-    datasets \
-    huggingface-hub "protobuf<4" "click<8.1"
+    python -m venv $HOME/.venv
+
+ENV PATH="$HOME/.venv/bin:$PATH"
+RUN echo 'source $HOME/.venv/bin/activate' >> $HOME/.bashrc
 
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+RUN pip install --no-cache-dir --upgrade pip setuptools wheel
+
+WORKDIR /tmp
+COPY --chown=user:user ./requirements.txt /tmp/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /tmp/requirements.txt
+
+RUN pip install InvokeAI --use-pep517 --extra-index-url https://download.pytorch.org/whl/cu121
 
-# Set the working directory to /data if USE_PERSISTENT_DATA is set, otherwise set to $HOME/app
 WORKDIR $HOME/app
-EXPOSE 9090
 
+ENV INVOKEAI_ROOT=${USE_PERSISTENT_DATA:+/data/invokeai}
+ENV INVOKEAI_ROOT=${INVOKEAI_ROOT:-$HOME/invokeai}
 
-COPY --chown=user:user ./invokeai.yaml /home/user/invokeai/invokeai.yaml
+COPY --chown=user:user ./start.sh $HOME/start.sh
+COPY --chown=user:user invokeai.yaml $HOME/invokeai.yaml
 
-# RUN ["invokeai-configure", "--yes", "--default_only", "--root", "."]
+RUN chmod +x $HOME/start.sh
 
-# Install models according to https://github.com/invoke-ai/InvokeAI/blob/main/docs/installation/050_INSTALLING_MODELS.md
-# RUN ["invokeai-model-install", "--add","https://civitai.com/api/download/models/128713"]
+EXPOSE 9090
 
-# Run IDE
-CMD ["invokeai-web"]
+ENTRYPOINT ["/bin/bash", "-c"]
+CMD ["/home/user/start.sh"]