通过API运行Stable Diffusion的简明指南

通过API运行Stable Diffusion

Stable Diffusion开源的一个好处在于，可以修改它并用它来构建各种应用，例如Photoshop插件、机器人、动画，以及修复人类的瑕疵等等。

但是，如果想将其集成到应用程序或项目中，就需要用GPU来设置它，并在其前面放置一个API。而GPU价格昂贵，因此不会希望一直让它们处于开启状态。

某平台允许您从自己的代码中运行机器学习模型，而无需设置任何基础设施。本文将展示如何使用它来运行Stable Diffusion。

安装Python库

该平台为API维护一个开源的Python客户端。使用pip安装它：

pip install replicate

还有一个社区维护的Node.js/JavaScript库。详情请见GitHub上的replicate-js。

身份验证

注册某平台，然后可以在您的账户页面上找到您的API令牌。在终端shell中将令牌设置为环境变量：

export REPLICATE_API_TOKEN=<token>

您可以免费使用API一段时间，但最终会要求您输入信用卡信息。费用仅根据您的请求运行时间按秒计费，因此通常比运行自己的GPU要便宜得多。

运行预测

创建一个名为dream.py的文件，并粘贴以下内容：

import replicate
import webbrowser

model = replicate.models.get(“stability-ai/stable-diffusion”)
version = model.versions.get(“db21e45d3f7023abc2a46ee38a23973f6dce16bb082a930b0c49861f96d1e5bf”)
output_url = version.predict(prompt=“electric sheep, neon, synthwave”)[0]
print(output_url)
webbrowser.open(output_url)

然后从终端运行脚本：

python dream.py

这将使用API创建一个预测，并在您的网络浏览器中打开生成的图像。效果可能类似于这样：

查看您的预测

无论您是在浏览器中还是通过API在某平台上运行模型，预测都会被保存并与您的用户账户关联。访问您的仪表板可以查看您之前所有的预测。

后续步骤

• 学习如何使用Stable Diffusion构建Discord机器人。
• 查看更多可以构建的示例。
• 尝试设置init_image参数以进行图像到图像的生成。
• 了解更多关于HTTP API的工作原理。
• 使用andreasjansson/stable-diffusion-animation生成动画。

欢迎加入我们的Discord，向我们展示您的作品，或者如果您需要任何帮助。我们期待看到您的创作。✨

下一篇： 使用Stable Diffusion、某平台和Fly.io为您的Discord服务器构建一个机器人艺术家。