merge

Signed-off-by: fbt <fbt@fleshless.org>

Containerfile

@@ -8,7 +8,7 @@ RUN git clone https://aur.archlinux.org/stable-diffusion-ui.git /sdu
 
 WORKDIR /sdu
 RUN chown -R build:build /sdu
-RUN sudo -u build makepkg -is --noconfirm
+RUN sudo -u build makepkg -s --noconfirm
 RUN mv -v stable-diffusion-ui*.pkg.tar.zst /stable-diffusion-ui.pkg.tar.zst
 
 # Stage 1
@@ -16,12 +16,9 @@ FROM archlinux:latest
 COPY --from=0 /stable-diffusion-ui.pkg.tar.zst /stable-diffusion-ui.pkg.tar.zst
 
 RUN pacman-key --init
-RUN pacman -Suy --noconfirm
+RUN pacman -Suy --noconfirm which nvidia-utils
 
 COPY init /init
 COPY config.yaml /config.yaml
 
-RUN pacman -S --noconfirm which nvidia-utils
-RUN useradd -m -s /usr/bin/bash -d /srv/stable-diffusion stable-diffusion
-
-ENTRYPOINT /init
+CMD /init

README.md

@@ -2,6 +2,8 @@
 
 This repo is an easy way to run Easy Diffusion in a podman container. I needed to put Easy Diffusion into a container for my own purposes, and publishing this wasn't a lot of effort; so here we are.
 
+I highly recommend rootless podman, as otherwise running the container will result in the ui running as root in a privileged container.
+
 ~~This targets Nvidia and Arch, and nothing else, I'm old and lazy.~~
 
 It should work on anything that has an Nvidia GPU, podman, and the nvidia container toolkit (with the nvidia devices setup hook).
@@ -15,10 +17,10 @@ Note that there's currently an issue with the AUR packages for Nvidia's containe
 
 ```
 # Build the image
-podman build --no-cache . -t 'stable-diffusion-ui-server'
+podman build . -t 'stable-diffusion-ui'
 
 # Run the container
-podman run -it --device nvidia.com/gpu=all --security-opt=label=disable --name stable-diffusion-ui -p 9000:9000 -v ./data:/opt/stable-diffusion-ui -e NVIDIA_DRIVER_CAPABILITIES=compute,utility localhost/stable-diffusion-ui-server
+podman run -it --device nvidia.com/gpu=all --security-opt=label=disable --name stable-diffusion-ui -p 9000:9000 -v ./data:/opt/stable-diffusion-ui localhost/stable-diffusion-ui
 
 # Get to the interface
 browser http://localhost:9000

config.yaml

@@ -9,7 +9,7 @@ render_devices: auto
 
 # Set open_browser_on_start to false to disable opening a new browser tab on each restart
 ui:
-  open_browser_on_start: false
+  open_browser_on_start: true
 update_branch: main
 
 # Set force_save_path to enforce an auto save path. Clients will not be able to change or
@@ -22,6 +22,6 @@ update_branch: main
 use_v3_engine: true
 models_dir: /opt/stable-diffusion-ui/models
 model:
-  stable-diffusion: lazyphoton-0.50000
+  stable-diffusion: realism/photon_v1
   vae: vae-ft-mse-840000-ema-pruned
 vram_usage_level: low

init

@@ -5,6 +5,8 @@ cfg="/opt/stable-diffusion-ui/config.yaml"
 
 if [[ -f "$pkg" ]]; then
 	pacman -U --noconfirm --overwrite '/opt/stable-diffusion-ui/*' "$pkg"
+	rm -vf "$pkg"
+fi	
 
 if ! [[ -f "$cfg" ]]; then
 	mv /config.yaml "$cfg" || {
@@ -12,8 +14,4 @@ if [[ -f "$pkg" ]]; then
 	}
 fi
 
-	chown -R stable-diffusion:stable-diffusion /opt/stable-diffusion-ui
-	rm -vf "$pkg"
-fi
-
-exec su stable-diffusion -c '/usr/bin/stable-diffusion-ui-server'
+exec /usr/bin/stable-diffusion-ui-server