Update Dockerfile

Dockerfile

@@ -1,7 +1,9 @@
 FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
-FROM python:3.10.12
-# WORKDIR /code
 
+ENV DEBIAN_FRONTEND=noninteractive \ 
+    TZ=America/Los_Angeles
+    
+ARG USE_PERSISTENT_DATA
 
 RUN apt-get update && apt-get install -y \
     git \
@@ -11,19 +13,48 @@ RUN apt-get update && apt-get install -y \
     ffmpeg libsm6 libxext6 cmake libgl1-mesa-glx \
     && rm -rf /var/lib/apt/lists/* \
     && git lfs install
-    
 
-RUN git clone https://github.com/lllyasviel/Fooocus.git
+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
 
-WORKDIR /Fooocus
+ARG PYTHON_VERSION=3.10.12
+# 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"
 
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
-# RUN cd Fooocus
-RUN python3 -m venv fooocus_env
-# RUN source fooocus_env/bin/activate
-RUN pip install -r requirements_versions.txt
+# Set the working directory to /data if USE_PERSISTENT_DATA is set, otherwise set to $HOME/app
+WORKDIR $HOME/app
 
+# Copy the current directory contents into the container at $HOME/app setting the owner to the user
+
+RUN git clone https://github.com/lllyasviel/Fooocus . && \
+    # pin to a specific commit
+    git checkout fc3588875759328d715fa07cc58178211a894386 && \
+    pip install --no-cache-dir xformers==0.0.20 triton==2.0.0 torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/cu118 && \
+    pip install --no-cache-dir -r requirements_versions.txt
+
+# remove the next line if you're running on your own GPU, it set max images to 3 and disables the API
+RUN sed -i "s|image_number = gr.Slider(label='Image Number', minimum=1, maximum=32|image_number = gr.Slider(label='Image Number', minimum=1, maximum=3|" webui.py && \
+    sed -i "s|shared.gradio_root = gr.Blocks(title='Fooocus ' + fooocus_version.version, css=modules.html.css).queue()|shared.gradio_root = gr.Blocks(title='Fooocus ' + fooocus_version.version, css=modules.html.css).queue(concurrency_count=1,api_open=False)|" webui.py && \
+    sed -i "s|shared.gradio_root.launch(inbrowser=True, server_name=args.listen, server_port=args.port, share=args.share)|shared.gradio_root.launch(inbrowser=True, server_name=args.listen, server_port=args.port, share=args.share, show_api=False)|" webui.py
 
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH \
@@ -32,9 +63,7 @@ ENV HOME=/home/user \
     GRADIO_ALLOW_FLAGGING=never \
     GRADIO_NUM_PORTS=1 \
     GRADIO_SERVER_NAME=0.0.0.0 \
-    GRADIO_SERVER_PORT=7860 \
     GRADIO_THEME=huggingface \
     SYSTEM=spaces
-    
-# RUN source fooocus_env/bin/activate
-RUN python entry_with_update.py --listen --port 7860 --always-cpu --share
+
+CMD ["python", "launch.py"]