AI 日报

如何运行新的 2.1 稳定扩散模型

  • By aihubon
  • Dec 19, 2023 - 2 min read



如何运行新的 2.1 稳定扩散模型

我在 Reddit 的 Stable Diffusion subreddit 上看到一些人很难获得新的 2.1 模型设置。这是新模型设置的快速指南自动1111(它也应该适用于大多数其他网络用户界面)。

在开始之前,请确保 AUTOMATIC1111 已更新到最新版本。您可以通过打开安装目录、打开 CMD 并运行以下命令来完成此操作:

git pull

现在您已经获得了最新版本,如果您还没有下载 v2.1 检查点文件,请从 HuggingFace 下载。

https://huggingface.co/stabilityai/stable-diffusion-2-1/tree/main

下载“v2-1_768-nonema-pruned.ckpt”版本并将其放入稳定扩散模型文件夹中。除非您打算训练自己的模型,在这种情况下您应该下载 ema 版本。

您需要的下一个文件是 .yaml 配置。下载此文件的最简单方法是右键单击 GitHub 链接并另存为,确保将其保存为 .yaml 文件而不是 .txt。

https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.yaml

将 .yaml 文件重命名为“v2-1_768-nonema-pruned.yaml”,并将其移至之前放置 .ckpt 文件的模型文件夹中。

现在您应该能够使用 AUTOMATIC1111 启动稳定扩散并选择新的 v2.1 模型作为检查点。如果您的显卡较弱并且难以生成图像,您可能需要使用以下内容编辑 webui-user.bat 文件。

代替:

set COMMANDLINE_ARGS=

和:

set COMMANDLINE_ARGS=--precision full --no-half --medvram

这应该会释放一些内存并阻止稳定扩散占用大量资源,但生成图像可能需要更长的时间。看来2.0和2.1型号比1.4和1.5更难跑。