AI绘画：搭建自己的AI绘画网站

作者：华王
星球：https://t.zsxq.com/0dgMjetVg (opens new window)

学习、分享、成功；提高效率，有所收获！😄

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

接下来小编带搭建从零搭建一个自己的AI绘画网站，基于本地模型来进行AI绘画操作

参考：

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

准备

硬件需求：

必须要有NVIDIA GPU，至少4GB VRAM，本地磁盘至少有15GB的空间，我们打包的项目解压后需要8.1G的磁盘。（模型比较大）

软件安装：

1 需要安装 Python 3.10.6 2 安装git 3 下载代码：git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git

python安装参考：

网址：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，地址开头已经写出

1 创建虚拟环境

conda create -n aiPicture python==3.10.6 conda activate aiPicture

2 查看自己cuda,安装对应torch

https://pytorch.org/get-started/locally/ 安装：pip3 install torch torchvision torchaudio

测试:

3 下载代码,安装依赖

git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui python -m pip install -r requriements.txt #进入仓库目录

4 模型下载

C站：https://civitai.com/

可以按照开头给的地址进行下载，下面以C站为例 进入网站后，搜索ChilloutMix（当然其他也可以），下载模型

注：模型文件有 2 种格式，分别是 .ckpt（Model PickleTensor） 和 .safetensors（Model SafeTensor），据说 .safetensors 更安全，这两种格式 stable-diffusion-webui 都支持，随意下载一种即可。

项目默认模型是这个，有需要可以去下载(启动执行webbui.sh 会自动下载)

https://huggingface.co/runwayml/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.ckpt

把模型放到工程目录：models/Stable-diffusion

启动网站

启动:"python launch.py --listen"，等待一定的时间出现网页地址就可以了

或者执行：Run webui.sh 区别是会自动下载默认模型，因为我们已经自己下载好了模型，所以使用上面命令

遇到的问题

问题一：无法安装open_clip

这是因为无法链接到队形外网导致的

解决：pip install open_clip

问题二：无法下载库（RuntimeError: Couldn't clone Taming Transformers.）

解决：手动安装：>/stable-diffusion-webui/repositories$ git clone git@github.com:CompVis/taming-transformers.git

使用网站

打开网址，选择刚刚的生成模型

参考C站上的prompt生成

最后我们自己就可以用自己AI网站生成我们想要的图片了。