Stable Diffusion XL webui Linux服务器部署

本人把部署过程遇到的坑，会尽可能详细地写在这篇文章中，以供参考。希望能对大家有所帮助！

机器配置情况

4张RTX 4090，NVIDIA驱动版本为525.60.13，CUDA版本为12.0。

开始部署

我部署的SDXL版本为SDXL-refiner-1.0，虽然SDXL官方github提供了UI界面的脚本，但是用起来不如 AUTOMATIC1111开发的好用，所以还是推荐用AUTOMATIC1111的stable-diffusion-webui。

然后我用的是conda虚拟环境，python版本是3.10.13。切记python版本一定要3.10以上！

安装webui前准备

首先把stable-diffusion-webui克隆到服务器上

git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
cd stable-diffusion-webui

手动克隆依赖git仓库

由于网络原因，stable-diffusion-webui所依赖的5个仓库（stablediffusion、generative-models、k-diffusion、BLIP），基本不可能一次性自动下载下来，所以最好自己事先克隆好。
先把stable-diffusion-webui\modules\launch_utils.py中大概380行左右的5行代码注释掉：

    # git_clone(stable_diffusion_repo, repo_dir('stable-diffusion-stability-ai'), "Stable Diffusion", stable_diffusion_commit_hash)
    # git_clone(stable_diffusion_xl_repo, repo_dir('generative-models'), "Stable Diffusion XL", stable_diffusion_xl_commit_hash)
    # git_clone(k_diffusion_repo, repo_dir('k-diffusion'), "K-diffusion", k_diffusion_commit_hash)
    # git_clone(codeformer_repo, repo_dir('CodeFormer'), "CodeFormer", codeformer_commit_hash)
    # git_clone(blip_repo, repo_dir('BLIP'), "BLIP", blip_commit_hash)

再新建一个文件夹repositories，放在stable-diffusion-webui下，操作如下

mkdir repositories
cd repositories
git clone https://github.com/Stability-AI/stablediffusion.git stable-diffusion-stability-ai
git clone https://github.com/Stability-AI/generative-models.git
git clone https://github.com/crowsonkb/k-diffusion.git
git clone https://github.com/sczhou/CodeFormer.git
git clone https://github.com/salesforce/BLIP.git

这里说明一下：因为stable-diffusion-webui会把stablediffusion改成stable-diffusion-stability-ai，所以上述第一个克隆命令做了一下修改。

下载SDXL-refiner-1.0预训练模型

我这里用的是sd_xl_refiner_1.0.safetensors预训练模型权重，其他版本的可以去huggingface上下载，或者在官方git上打开相应版本的链接：
下载完以后，放在stable-diffusion-webui/models/Stable-diffusion/路径下

换pip源

由于webui提供的自动安装的shell脚本是基于pip来安装的，所以需要换成国内镜像源，来加快下载速度，命令如下：

pip config set global.index-url https://mirrors.aliyun.com/pypi/simple

安装

准备就绪后，先返回到stable-diffusion-webui文件夹下，然后运行webui.sh 文件

cd stable-diffusion-webui
./webui.sh

如果出现以下报错：

File "/stable-diffusion-webui/repositories/k-diffusion/k_diffusion/layers.py", line 4, in <module>
    from dctorch import functional as df
ModuleNotFoundError: No module named 'dctorch'

通过指定路径来安装，因为stable-diffusion-webui会创建一个venv文件夹，存放所有的依赖包。具体如下：

pip install dctorch --target /stable-diffusion-webui/venv/lib/python3.10/site-packages/

如果在终端输入./webui.sh以后，出现以下情况：

Launching Web UI with arguments:
no module 'xformers'. Processing without...
no module 'xformers'. Processing without...
No module 'xformers'. Proceeding without it.

也可以通过同样的方式来安装，不过需要注意的是，要与python版本、torch版本（通过webui.sh来安装的话就是2.0.1）相匹配，我这里安装的xformers是0.0.20 版本，

pip install xformers==0.0.20 --target /stable-diffusion-webui/venv/lib/python3.10/site-packages/

xformers是用来加速MHA计算的，使用xformers包的话可以用以下操作：

./webui.sh --xformers

出现下面的URL，就表示安装完成了！

设置public link

由于我是部署在服务器，有时候想windows本地使用的话，需要能外部访问，所以想设置一个外部链接。
首先，直接将命令改成：

./webui.sh --share --xformers

下载相关文件

如果有如下报错：

1. 首先手动下载frpc_linux_amd64文件
2. 然后按要求重命名为frpc_linux_amd64_v0.2
3. 最后放在/stable-diffusion-webui/venv/lib/python3.10/site-packages/gradio/文件夹下
4. 给frpc_linux_amd64_v0.2可执行权限：

chmod +x /stable-diffusion-webui/venv/lib/python3.10/site-packages/gradio/frpc_linux_amd64_v0.2

修改配置文件

同样在 /stable-diffusion-webui/venv/lib/python3.10/site-packages/gradio/文件夹下，找到networking.py，大概在28行左右：

# By default, the local server will try to open on localhost, port 7860.
# If that is not available, then it will try 7861, 7862, ... 7959.
INITIAL_PORT_VALUE = int(os.getenv("GRADIO_SERVER_PORT", "7860"))
TRY_NUM_PORTS = int(os.getenv("GRADIO_NUM_PORTS", "100"))
LOCALHOST_NAME = os.getenv("GRADIO_SERVER_NAME", "127.0.0.1")
GRADIO_API_SERVER = "https://api.gradio.app/v2/tunnel-request"

将LOCALHOST_NAME改成自己服务器的IP，也就是下面这行：

LOCALHOST_NAME = os.getenv("GRADIO_SERVER_NAME", "YOUR IP")

相应地，端口也可以在INITIAL_PORT_VALUE中修改。
接着，在121行左右，修改server_name为你服务器的IP：

server_name = server_name or LOCALHOST_NAME
url_host_name = "localhost" if server_name == "0.0.0.0" else server_name

改成

server_name = "YOUR IP"
url_host_name = "localhost" if server_name == "0.0.0.0" else server_name

至此，已经可以通过外部访问了。

设置账户密码

出于安全考虑，可以设置访问的账户和密码，只需在在运行命令的时候加上--gradio-auth user:password选项即可。冒号前面是你登录的账户名，后面是账户对应的密码。

最后运行：

./webui.sh --share --xformers --gradio-auth user:password

出现以下界面就大功告成了！撒花！
顺带一提：用自己服务器部署比在hugging face上直接用别人的快多了，按默认设置跑，3秒一张图，嘎嘎香！