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

11 阅读2分钟

通过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服务器构建一个机器人艺术家。