space setup permissions, etc.

Dockerfile

@@ -85,11 +85,11 @@ RUN git clone https://github.com/Mikubill/sd-webui-controlnet /app/stable-diffus
 # Prepare WebUI environment
 WORKDIR /app/stable-diffusion-webui
 COPY config.json ui-config.json /app/stable-diffusion-webui/
-RUN /opt/venv/bin/python launch.py --exit --skip-torch-cuda-test
+RUN /opt/venv/bin/python launch.py --exit --skip-torch-cuda-test --xformers
 
 RUN sed -i -e 's/                show_progress=False,/                show_progress=True,/g' modules/ui.py
 RUN sed -i -e 's/shared.demo.launch/shared.demo.queue().launch/g' webui.py
-RUN sed -i -e 's/ outputs=\[/queue=False, &/g' modules/ui.py
+# RUN sed -i -e 's/ outputs=\[/queue=False, &/g' modules/ui.py
 
 # Copy startup scripts
 COPY run.py on_start.sh /app/stable-diffusion-webui/
@@ -97,7 +97,7 @@ RUN chmod +x on_start.sh
 
 # Run app as non-root user
 RUN adduser --disabled-password --gecos '' user
-RUN chown -R user:user /app /opt/venv
+RUN chown -R user:user /app
 USER user
 
 EXPOSE 7860

on_start.sh

@@ -1,18 +1,6 @@
 #!/bin/bash
 set -euo pipefail
 
-echo "\$ nvidia smi"
-nvidia-smi
-
-# Install xformers matching the GPU (only for Nvidia A10G and T4 GPUs)
-if nvidia-smi --query-gpu=gpu_name --format=csv,noheader | grep -q A10G; then
-    /opt/venv/bin/pip install https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.16/xformers-0.0.16+814314d.d20230119.A10G-cp310-cp310-linux_x86_64.whl
-elif nvidia-smi --query-gpu=gpu_name --format=csv,noheader | grep -q T4; then
-    /opt/venv/bin/pip install https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.16/xformers-0.0.16+814314d.d20230118-cp310-cp310-linux_x86_64.whl
-else
-    /opt/venv/bin/pip install --pre xformers
-fi
-
 # SD 2.1 768 base model
 aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/stabilityai/stable-diffusion-2-1/resolve/36a01dc742066de2e8c91e7cf0b8f6b53ef53da1/v2-1_768-ema-pruned.safetensors -d /app/stable-diffusion-webui/models/Stable-diffusion -o v2-1_768-ema-pruned.safetensors
 aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://raw.githubusercontent.com/Stability-AI/stablediffusion/fc1488421a2761937b9d54784194157882cbc3b1/configs/stable-diffusion/v2-inference-v.yaml -d /app/stable-diffusion-webui/models/Stable-diffusion -o v2-1_768-ema-pruned.yaml