部署Stable-Diffusion
Stable Diffusion
stable-diffusion
是一种强大的文本到图像生成模型,它利用了潜在扩散模型、OpenCLIP编码器、超分辨率放大器等技术。 说白了就是AI绘画,目前非常火的项目,尤其是该项目是一个开源的项目,所以它可以独立部署在我们的机器上,AI出图也没有任何的限制。
配置要求
- 最低4GB显存显卡
- 16GB以上内存
环境
Python
安装Python (Python 3.10.6
为stable-diffusion
运行脚本所使用的测试版本,尽量下载3.10.6
版本)
安装完毕后,打开cmd
运行命令python
检查安装是否成功
python
-------------------
Python 3.11.3 (tags/v3.11.3:f3909b8, Apr 4 2023, 23:49:59) [MSC v.1934 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
获取Stable-Diffusion
SD github地址 release
中下载最新版本即可。
下载完毕后,将zip
压缩包进行解压
下载模型
模型下载可以在Civitai
进行下载,将下载的模型文件放在stable-diffusion
根目录下的/models/Stable-diffusion
目录中
推荐模型:ChilloutMix
下载模型时需要注意模型的类型,checkpoint
是大模型,lora
等其他模型都是微调的小模型,放入的目录不同。
启动Stable-Diffusion
配置
- 安装[
NVIDIA CUDA Toolkit
](https://developer.nvidia.com/cuda-toolkit-archive 作者:Reaper小冥 https://www.bilibili.com/read/cv22011634 出处:bilibili),cmd
运行以下命令,查询nvidia
版本,然后去官网下载相应的toolkit
安装包
nvidia-smi
- 修改根目录下的
webui-user.bat
文件
更多参考Command Line Arguments and Settings
@echo off
set PYTHON=
set GIT=
set VENV_DIR=
set CUDA_VISIBLE_DEVICES=1
set COMMANDLINE_ARGS= --xformers --lowvram
git pull
call webui.bat
--xformers
表示使用xformers
,可以提高绘制速度--lowvram
表示低显存,同时还有medvram
- 其他参数:
--precision full --no-half --skip-torch-cuda-test
用于Torch is not able to use GPU
报错 --no-gradio-queue
用于解决something went wrong
问题
启动
运行webui-user.bat
文件
第一次需要下载依赖,启动比较慢,启动成功如下图所示,会显示本地的网页地址:
运行报错解决方式
Torch is not able to use GPU
默认下载最新版的pytorch
,与本机的cuda
版本不匹配
cmd
运行nvidia-smi
命令,查看cuda
版本
- 删除根目录下的
venv
- 修改
launch.py
文件,搜索torch_command
#注释torch_command
#torch_command = os.environ.get('TORCH_COMMAND', "pip install torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/cu118")
#增加以下代码,其中下载命令可从官网寻找
torch_command = os.environ.get('TORCH_COMMAND', "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117")
要获取正确的版本需要访问pytorch
官网,点击install,可以按照自己的cuda选择
Couldn’t install gfpgan.
打开 stable-diffusion-webui/launch.py
文件。
找到 prepare_environment
方法:
在所有https://github.com
的链接前添加:https://ghproxy.com/
添加后的结果类似这样:https://ghproxy.com/https://github.com/....
重新运行webui-user.bat
即可。
版权声明:
本站所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自
Willxup!
喜欢就支持一下吧