使用 CPU 体验最顶尖的 AI 生成艺术模型

机器学习产生艺术的发展速度令人惊叹。它发生得如此之快，几乎让人感觉它就是凭空出现的。我们可以让计算机从单纯的文本提示中艺术性地创作出原创作品。Stable Diffusion 毫无疑问成为了当前最出名的生成艺术模型。它的效果堪称惊艳：

Midjourney + Stable Diffusion. Credit: alessandrochille, Darken, eyecon01

在周末的时间里我一直在把玩 Stable Diffusion，但是，它对于显存的需求也是非常之大。于是我希望它能够安静的跑在我 2019 年的 XPS 笔记本上，那么接下来就分享一下我的做法。

准备工作

首先，我确定使用 envd 来配置环境，开发环境通过 envd 运行在容器中，因此在尝试 Stable Diffusion 的过程中不会破坏我的笔记本电脑上已经脆弱不堪的本地环境。envd 可以通过 pip 进行安装：

pip3 install --pre --upgrade envd

接下来，我选择了 Huggingface 上的 Stable Diffusion 托管模型，因此需要：

直接运行

接下来，我们只需要运行，就可以了。让我们从配置开发环境开始：

bash

mkdir stable-diffusion
cd stable-diffusion
touch build.envd

我们创建了一个目录 stable-diffusion，在其中创建了一个文件 build.envd，这是 envd 的配置文件：

python

def build():
    base(os="ubuntu20.04", language="python")
    #config.pip_index(url = "https://pypi.tuna.tsinghua.edu.cn/simple")
    install.python_packages([
        "torch",
        "transformers",
        "diffusers",
    ])

如果你在国内的话，可以添加上第三行 pip index 的配置，让环境的构建更快。接下来，让我们创建用户运行 Stable Diffusion 的 python 代码 main.py：

python

import random
import sys
import os

from diffusers import StableDiffusionPipeline

def dummy(images, **kwargs):
    return images, False

# Read prompt from command line
prompt = " ".join(sys.argv[1:])

pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4",
                                                use_auth_token=os.environ['HUGGINGFACE_TOKEN'])
pipe.to("cpu")
pipe.safety_checker = dummy

# Run until we exit with CTRL+C
while True:
    n = random.randint(1000, 9999)
    image = pipe(prompt, guidance_scale=7.5).images[0]
    image.save(f'{n}.jpeg')

接下来，我们通过 envd up 创建出开发环境，并且在其中设置环境变量 HUGGINGFACE_TOKEN 为你在准备工作中创建的 Huggingface token。最后，运行即可：

envd up
~/stable-diffusion via Py v3.9.13 via 🅒 envd 
⬢ [envd]❯ export HUGGINGFACE_TOKEN=<YOUR TOKEN HERE>
~/stable-diffusion via Py v3.9.13 via 🅒 envd 
⬢ [envd]❯ python main.py "your prompt here"

下面是我自己通过 Stable Diffusion 生成的图片，确实让人惊艳。起码，它比我更懂设计。

python main.py "a girl with lavender hair and black skirt, fairy tale style background, a beautiful half body illustration, top lighting, perfect shadow, soft painting, reduce saturation, leaning towards watercolor, art by hidari and krenz cushart and wenjun lin and akihiko yoshida"