synchronization

stable-diffusion-webui/Dockerfile

@@ -0,0 +1,20 @@
+FROM python:3.10.6
+
+LABEL maintainer="Offends <offends4@163.com>"
+
+RUN useradd webui \
+    && mkdir /home/webui -p \
+    && mkdir /stable-diffusion-webui/ -p \
+    && pip install --upgrade pip \
+    && apt update \
+    && apt install libgl1-mesa-glx -y \
+    && pip install opencv-python-headless
+
+WORKDIR /stable-diffusion-webui/
+
+EXPOSE 7860
+
+COPY ./start.sh /start.sh
+COPY ./builder.sh /builder.sh
+
+CMD ["bash","/start.sh"]

stable-diffusion-webui/README.md

@@ -0,0 +1,17 @@
+> 本文作者：丁辉
+
+# Stable-diffusion-webui-docker
+
+1. 构建基础镜像
+
+   ```bash
+   docker build -t webui:v1 .
+   ```
+
+2. 启动容器初始化文件
+
+   ```bash
+   docker run -it --name webui -p 7860:7860 -v /root/stable-diffusion-webui:/stable-diffusion-webui webui:v1 bash /start.sh
+   ```
+
+   

stable-diffusion-webui/builder.sh

@@ -0,0 +1,12 @@
+#!/bin/bash
+
+# 拉去文件
+cd /stable-diffusion-webui
+git init
+git remote add origin https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
+git pull origin master
+# 授权
+chown webui:webui -R /home/webui/
+chown webui:webui -R /stable-diffusion-webui/
+# 初始化文件
+su - webui -s /bin/bash -c "/stable-diffusion-webui/webui.sh --precision full --no-half --listen --skip-torch-cuda-test"

stable-diffusion-webui/start.sh

@@ -0,0 +1,29 @@
+#!/bin/bash
+
+# 判断 /stable-diffusion-webui 目录是否存在
+if [ ! -d "/stable-diffusion-webui" ]; then
+  echo "创建 /stable-diffusion-webui 目录"
+  mkdir /stable-diffusion-webui -p
+  chown webui:webui -R /stable-diffusion-webui/
+fi
+
+# 判断 /stable-diffusion-webui 目录下是否有文件
+
+if [ ! "$(ls -A /stable-diffusion-webui)" ]; then
+    echo "################################################################"
+    echo "正在准备拉取文件,文件大小10G, 预计10分钟左右,请稍等..."
+    echo "################################################################"
+    cd /
+    # wget ${OSS_URL:-"https://rainbond-pkg.oss-cn-shanghai.aliyuncs.com/rainstore/stable-diffusion-webui.tar.gz"}
+    wget https://rainbond-pkg.oss-cn-shanghai.aliyuncs.com/rainstore/stable-diffusion-webui-v15.tar.gz
+    tar -zxvf stable-diffusion-webui-v15.tar.gz -C /
+    echo "################################################################"
+    echo "文件准备完成"
+    echo "################################################################"
+fi
+
+echo "启动服务"
+# CPU 启动
+# su - webui -s /bin/bash -c "/stable-diffusion-webui/webui.sh --precision full --no-half --listen --skip-torch-cuda-test"
+# GPU启动
+su - webui -s /bin/bash -c "/stable-diffusion-webui/webui.sh --precision full --no-half --listen"