搭建自己的AI绘画网站(基于Stable Diffusion模型)

Stable Diffusion模型发布以来，"AI文本图片生成"进入人们视野。AI 绘画通用算法Stable Diffusion已然超神，最近一些网友将真人图片输入模型进行自主学习，这使得其训练效果已经可以以假乱真，人们很难分辨出哪些图片是通过AI生成的还是真人拍摄的。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

接下来小编带搭建从零搭建一个自己的AI绘画网站，基于本地模型来进行AI绘画操作文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

参考：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

git地址：
https://github.com/AUTOMATIC1111/stable-diffusion-webui
文档Doc:
https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki
模型地址：
1 huggingface
https://huggingface.co/CompVis/stable-diffusion
2 C站：
https://civitai.com/
3 默认模型
https://huggingface.co/runwayml/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.ckpt文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

准备

硬件需求：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

必须要有NVIDIA GPU，至少4GB VRAM，本地磁盘至少有15GB的空间，我们打包的项目解压后需要8.1G的磁盘。（模型比较大）文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

软件安装：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

python安装参考：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

网址：https://conda.io/en/latest/miniconda.html
wget https://repo.anaconda.com/miniconda/Miniconda3-py310_23.1.0-1-Linux-x86_64.sh #下载脚本
sh Miniconda3-py39_4.12.0-Linux-x86_64.sh # 执行
~/miniconda3/bin/conda init #初始化Shell，以便直接运行conda
conda create --name d2 python=3.9  #关启shell，创建虚拟环境
conda activate d2 #激活 d2环境

搭建AI网站

直接使用GitHub仓库：stable-diffusion-webui，地址开头已经写出文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

1 创建虚拟环境文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

conda create -n aiPicture python==3.10.6
conda activate aiPicture文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

2 查看自己cuda,安装对应torch文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

https://pytorch.org/get-started/locally/
安装：pip3 install torch torchvision torchaudio文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

测试:文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

3 下载代码,安装依赖文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui
python -m pip install -r requriements.txt #进入仓库目录文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

4 模型下载文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

C站：https://civitai.com/文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

可以按照开头给的地址进行下载，下面以C站为例
进入网站后，搜索ChilloutMix（当然其他也可以），下载模型文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

注：模型文件有2种格式，分别是.ckpt（Model PickleTensor）和 .safetensors（Model SafeTensor），据说 .safetensors 更安全，这两种格式 stable-diffusion-webui 都支持，随意下载一种即可。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

项目默认模型是下面这个，有需要可以去下载(启动执行webbui.sh 会自动下载)文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

https://huggingface.co/runwayml/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.ckpt文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

把模型放到工程目录：models/Stable-diffusion文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

启动网站

启动:"python launch.py --listen"，等待一定的时间出现网页地址就可以了文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

或者执行：Run webui.sh 区别是会自动下载默认模型，因为我们已经自己下载好了模型，所以使用上面命令文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

遇到的问题

问题一：无法安装open_clip文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

这是因为网络无法链接到外网导致的文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

解决：pip install open_clip文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

问题二：无法下载库（RuntimeError: Couldn't clone Taming Transformers.）文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

使用网站

打开网址，选择刚刚的生成模型文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

参考C站上的prompt生成文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

最后我们自己就可以用自己AI网站生成我们想要的图片了。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html

文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/34447.html