在本地运行稳定扩散【2023年12月版本】
在去年的圣诞倒数日历活动中,我写了使用M1 MacBook Pro进行环境配置的方法。
我想谈谈今年以来环境发生了相当大的变化。
前提 tí)
所需之物
-
- そこそこ早いCPU
-
- 出来るだけグラフィックメモリーが多いGPU(NVIDIA推奨)
-
- そこそこのメモリ
-
- そこそこのデータドライブ
-
- WSL 2(Windowsの場合)
-
- Docker
- Git
我目前所使用的环境
Windows – 窗户
-
- CPU:AMD Ryzen 9 5950X
-
- GPU:NVIDIA GeForce RTX 3080 Ti(12GB)
- メモリ:64GB
苹果电脑
M1 Max MacBook Pro 去年相比没有改变。
环境建设
本次环境构建将使用Docker。
去年我直接在机器上安装了Python并进行了配置,但现在我可以在Docker上创建,因此我将总结一下整个流程。
直接安装Python也是可以的,但对于每个项目需要切换版本时,使用类似Docker的虚拟化技术更加方便易用。
除非特别注明,否则无论是Windows还是Mac都需要进行相同的操作。
首先是整体流程。
-
- WSL 2设置(仅适用于Windows)
-
- Docker环境配置
- Stable Diffusion Web UI Docker设置
我会逐一查看。
设置WSL 2(仅适用于Windows)
需要在Windows上运行Docker,因此需要设置WSL 2。
只需要以管理员权限打开PowerShell或命令提示符,执行以下命令即可。
wsl --install
搭建Docker环境。
接下来是Docker。将使用Docker桌面版。
请打开页面,下载并安装。
稍微调整一下设置(适用于Windows)。
如果安装成功,我会启动它。
打开设置,确认“使用基于WSL 2的引擎”的复选框是否被勾选上了。如果没有勾选上,就勾选上它。
接下来,我们要进行.wslconfig文件的编辑。
打开C:\ Users \ \ .wslconfig。
我把我的设置放在这里。
[wsl2]
memory=16GB
swap=0
内存为总容量的一半或者是8GB中较小的那个值作为初始值。
如果像我一样装备了更高容量的内存,建议将此值更改为16GB以上。
如果您的Mac有足够的内存空间,我认为增加内存是一个不错的选择。
关于交换空间,它会因为使用C盘做什么而有所不同。由于我使用的是SSD,我不希望交换空间过多地使用,所以我将其设置为0。
准备工作到此为止。
稳定的扩散Web UI Docker的设置
现在,我们进入正题。
去年我们使用命令行界面来使用Stable Diffusion,但现在使用AUTOMATIC1111的Web用户界面是标准做法。
这是经过改编以在Docker上运行的稳定扩散WebUI Docker,我们将在本次使用它。
需要用git clone来使用。
git clone https://github.com/AbdBarho/stable-diffusion-webui-docker.git
另外,如果是在Windows系统中,想要利用GPU生成图像,则需要安装CUDA Toolkit。
只需执行以下命令即可启动Web UI。
docker compose --profile download up --build
docker compose --profile auto up --build
在Mac上,它不会自动工作,需要启动auto-cpu。
docker compose --profile download up --build
docker compose --profile auto-cpu up --build
可能是与GPU相关的问题。
一旦启动后,访问 http://localhost:7860/。
如果无法启动,请尝试重新启动机器。
如果到达这一步,之后只需要输入提示(咒语)就可以生成了。
总结而言
一旦设置完成后,将显示这样的界面。请尽情生成各种内容。
我在画面上尝试使用我常用的「一张在沙漠中骑马的宇航员的照片」作为提示生成图像。
由于实际上安装了许多扩展功能,所以界面可能略有不同。我希望能在另一个机会再次介绍这些内容。
祝你好好享受图像生成的乐趣!
額外禮品
对于模特来说
由于稳定扩散所使用的标准模型实际上并不太理想,因此我们需要根据想要呈现的画面来选择合适的模型。
模型正在逐渐形成,并集中在名为Civitai的网站上。
使用搜索功能,可以找到符合所需图片形象的模特。
顺便提一句,如果要提到包括日本人在内的亚洲人,有一个名为美丽真实亚洲模特(BRA)的模特组织很有名。
需要从下载中下载safetensors文件,并将其放置在Stable Diffusion Web UI Docker加载的位置。
使用Docker桌面版,在启动Docker容器的状态下,只需将已下载的模型文件拖放到/data/models/Stable-diffusion目录即可。
然后,从稳定扩散的Web UI界面上重新加载模型。