Skip to content

title: Stable Diffusion部署 date: 2023-4-4 18:13:58 tags: - python - pytorch categories: - [机器学习] excerpt: Windows和Ubuntu下的部署方法

Stable Diffusion部署

分别介绍了在window下和Ubuntu服务器下的部署方法

本地(Windows)

直接b站下整合包即可。 https://www.bilibili.com/video/BV1ne4y1V7QU

服务器(Ubuntu 18.04)

参考Ubuntu下配置pytorch环境

1.conda新建虚拟环境,注意python版本

conda create -n sd python=3.10.6
conda activate sd

接下来根据服务器是否能科学上网有两种方法

有梯子

  1. git clone 

    git config --global http.sslVerify false
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
cd ./stable-diffusion-webui

  2. 安装相关依赖

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113
pip install -r requirements_versions.txt
pip install -r requirements.txt
4. 启动(第一次的时候还要安装几个依赖的包)

python launch.py

无梯子

如果没有梯子,第3,4步可能会卡住。我们需要手动换源安装。参考的是知乎专栏

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113 --trusted-host download.pytorch.org -i https://pypi.doubanio.com/simple

pip install -r requirements_versions.txt --prefer-binary -i https://pypi.douban.com/simple/

pip install git+https://gitee.com/hznn/GFPGAN.git --prefer-binary
pip install git+https://gitee.com/hznn/CLIP.git --prefer-binary
pip install git+https://gitee.com/jerrylinkun/DeepDanbooru.git --prefer-binary


git clone https://gitee.com/jerrylinkun/stable-diffusion.git ./repositories/stable-diffusion


git clone https://github.com/CompVis/taming-transformers.git ./repositories/taming-transformers
git clone https://github.com/crowsonkb/k-diffusion.git ./repositories/K-diffusion


git clone https://gitee.com/arcsion/CodeFormer.git ./repositories/CodeFormer
git clone https://github.com/salesforce/BLIP.git ./repositories/BLIP

pip install -r ./repositories/CodeFormer/requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install opencv-python-headless -i https://pypi.douban.com/simple/

pip install git+https://gitee.com/ufhy/open_clip.git --prefer-binary

然后再执行

python launch.py

即可

使用

每次启动前别忘了激活环境

conda activate sd
python launch.py

Lora安装

进入WebUI界面找到Settings/Extensions, 安装sd-webui-additional-networks. 如果因为梯子的原因找不到，可以直接点Install from URL 然后填入https://ghproxy.com/https://github.com/kohya-ss/sd-webui-additional-networks.git

我们在civitai等网站上下载模型后要放到指定目录。注意放模型的位置不是 /models/Lora而是在 /extensions/sd-webui-additional-networks/models/lora

重启UI之后，应该能看到Additional Network这一栏，然后勾选Enable

然后再选择模型和权重

Comments