Merge 208405e8cc into 802d0bcd68

services/AUTOMATIC1111/Dockerfile

@@ -50,6 +50,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
 RUN apt-get -y install libgoogle-perftools-dev && apt-get clean
 ENV LD_PRELOAD=libtcmalloc.so
 
+
 COPY . /docker
 
 RUN \
@@ -58,6 +59,33 @@ RUN \
   sed -i 's/in_app_dir = .*/in_app_dir = True/g' /opt/conda/lib/python3.10/site-packages/gradio/routes.py && \
   git config --global --add safe.directory '*'
 
+# After the base Python/Conda setup, add:
+RUN pip install insightface
+# Fix numpy compatibility issues
+RUN pip install --no-cache-dir \
+    numpy==1.23.5 \
+    scikit-image==0.19.3 \
+    opencv-python-headless==4.8.1.78
+
+# Install system dependencies for ControlNet
+RUN apt-get update && apt-get install -y \
+    libglib2.0-0 \
+    libsm6 \
+    libxext6 \
+    libxrender-dev \
+    libgomp1 \
+    libgstreamer1.0-0 \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install mediapipe for ControlNet
+RUN pip install --no-cache-dir mediapipe==0.10.9
+
+# Pre-install ControlNet (optional - if you want it built-in)
+RUN cd /stable-diffusion-webui/extensions && \
+    git clone https://github.com/Mikubill/sd-webui-controlnet.git && \
+    cd sd-webui-controlnet && \
+    pip install --no-cache-dir -r requirements.txt
+
 WORKDIR ${ROOT}
 ENV NVIDIA_VISIBLE_DEVICES=all
 ENV CLI_ARGS=""