使用稳定扩散等技术为Discord服务器构建艺术机器人
使用稳定扩散等技术为Discord服务器构建艺术机器人
去年VQGAN-CLIP发布后不久，某机构Discord服务器上的#the-faraday-cage频道里就有人制作了一个运行该模型的机器人。你要求它 !imagine 某个东西，然后VQGAN-CLIP就会用它那奇特的方式诠释你的请求。
这非常令人兴奋。这是文本到图像模型的第一个游乐场，是人们第一次可以提示计算机创作艺术的地方。而且，它非常有趣！模型不再被锁在Colab笔记本里，而是像一百个人在同一个房间里一起玩耍。人们互相借鉴想法、互相鼓励、对彼此的输出结果开怀大笑。“虚幻引擎”、“艺术站”等提示词技巧就是在这里出现的。有人看到别人做了些很酷的事情，然后提示词的“迷因”便传播开来。
自从那些四只耳朵、没有脸的兔子图像出现以来，我们已经走过了很长的路。本周早些时候，我们很幸运地获得了来自“某大型A100集群”的模型权重。那么，让我们好好利用它们。
本指南将展示如何用Stable Diffusion创建你自己的Discord机器人，这样你就可以在自己的Discord服务器里拥有一个强大得多的机器人艺术家了。
我们将使用某中心的平台来运行模型，因此你不需要拥有GPU，同时我们会将其部署到Fly.io上，这样即使你的电脑关机，它也能保持全天候运行。
先决条件
- 安装 Python 3.5 或更高版本。
- 一个命令行终端（例如 Mac 上的 Terminal，Linux 上的 gnome-terminal，或 Windows 上的 PowerShell）。
- 在某中心平台注册一个账户。
- 在 Discord 注册一个账户。
- 阅读模型许可协议，以便了解你可以将该模型用于哪些用途。
设置一个 Discord 机器人账户
遵循 discord.py 的指南来设置一个 Discord 机器人账户。给它起个名字，比如 stable-diffusion-bot。在第一部分结束时，你需要在机器人的设置页面开启“消息内容意图”选项。你的机器人需要这个权限来读取消息内容，以获取提示词！
在该指南的后半部分（“邀请你的机器人”）中，当它告诉你“勾选你的机器人运行所需的权限”时，你需要给予机器人“发送消息”的权限。
编写代码
本节将引导你完成设置代码库所需的步骤。
🐇 想跳过步骤？如果你不想复制粘贴下面所有的代码，可以直接克隆 replicate/discord-bot-tutorial-code 这个 GitHub 仓库，从一个可工作的项目模板开始。
创建项目目录
mkdir stable-diffusion-bot
cd stable-diffusion-bot
定义 Python 依赖项
接下来，你需要定义项目所需的一些 Python 包依赖。
创建一个名为 requirements.txt 的新文件，并粘贴以下内容：
discord.py==2.0.1
packaging==21.3
python-dotenv==0.20.0
replicate==0.0.1a15
然后安装依赖项：
pip install -r requirements.txt
配置环境和密钥
创建一个名为 .env 的文件。这个文本文件将用于存储开发环境的密钥。粘贴以下内容：
REPLICATE_API_TOKEN=<你的令牌>
DISCORD_TOKEN=<你的令牌>
访问某中心的账户页面以复制粘贴你的 API 令牌。
Discord 令牌是你在按照 Discord 指南创建机器人时复制的令牌。如果没有，可以访问 Discord 开发者门户，选择你的机器人应用，从侧边栏菜单中选择“Bot”，然后点击“Reset Token”来生成新令牌。
✋ .env 文件包含密钥，因此不应与任何人共享。如果你计划将项目转为 Git 仓库，请务必创建一个 .gitignore 文件并将 .env 添加进去。
编写机器人代码
创建一个名为 bot.py 的新文件，并将以下代码粘贴进去：
from discord import Intents
from discord.ext import commands
from dotenv import load_dotenv
import os
import replicate
load_dotenv()
intents = Intents.default()
intents.message_content = True
bot = commands.Bot(
command_prefix="!",
description="在某中心上运行模型！",
intents=intents,
)
@bot.command()
async def dream(ctx, *, prompt):
"""使用 stable-diffusion 模型根据文本提示生成图像"""
msg = await ctx.send(f"“{prompt}”
> 生成中...")
model = replicate.models.get("stability-ai/stable-diffusion")
version = model.versions.get("27b93a2413e7f36cd83da926f3656280b2931564ff050bf9575f1fdf9bcd7478")
image = version.predict(prompt=prompt)[0]
await msg.edit(content=f"“{prompt}”
{image}")
bot.run(os.environ["DISCORD_TOKEN"])
这个文件定义了你的机器人命令 (!dream) 及其工作原理。以后想要修改机器人时，就需要回到这个文件。
在本地运行机器人
编写完机器人代码后，是时候运行它了！
运行以下命令在本地启动你的机器人：
python bot.py
使用机器人
在你的 Discord 服务器的任意频道中输入 !help，检查你的机器人是否正在运行。你的机器人应该会回应它能运行的命令列表，包括 !dream。
现在尝试生成一张图像：
!dream 草地上的兔子，artstation trending
你的机器人应该会写一条“生成中…”的消息，大约 20-40 秒后，它会用新生成的图像替换那条消息。🥳
使用 Fly.io 部署机器人
在本地运行机器人很方便，尤其是在你积极开发它的时候。然而，缺点是它只能在你在线时运行。如果你是为一个服务器构建机器人，你可能希望即使你不在线时它也能运行。
部署此类应用程序的方法有很多。最简单的是使用诸如 Vercel、Fly.io 或 Heroku 之类的服务。在本教程中，你将使用 Fly 将机器人部署到云端。
首先，请参考 Fly 的“快速入门”指南，安装 flyctl 命令行工具并创建一个 Fly 账户。
然后创建一个名为 Dockerfile 的新文件，并将以下代码粘贴进去：
FROM python:3.10
WORKDIR /code
COPY requirements.txt /code/
RUN pip install -r requirements.txt
COPY . /code
CMD python bot.py
然后创建一个新的 Fly 应用：
flyctl launch
该命令将生成一个名为 fly.toml 的新文件，但它是为运行 Web 应用设计的，因此你需要做一些修改。删除 [[services]] 块及其下面的所有内容。修改后的文件应该类似这样：
# fly.toml 文件于 2022-08-25T17:37:31+01:00 为 your-fly-app 生成
app = "your-fly-app"
kill_signal = "SIGINT"
kill_timeout = 5
processes = []
[env]
[experimental]
allowed_public_ports = []
auto_rollback = true
然后使用本地 .env 文件中的密钥配置你的 Fly 应用：
flyctl secrets set REPLICATE_API_TOKEN=... DISCORD_TOKEN=...
就是这样！你的机器人现在已经在云端运行了。
如果你在服务器中运行这个机器人，有一件事需要注意，就是需要让使用它的人了解模型的许可协议。你可能想将其添加到服务器规则、频道主题或类似的地方。
定制你的机器人
现在可能是对机器人进行一些调整的好时机。一些想法：
- 生成多张图像而不是单张，并全部展示。
- 添加另一个命令来运行不同的模型。
- 获取图像输出并通过另一个模型（例如超分辨率模型）进行处理。
- 将其变成游戏。也许是“传话”游戏，或者“你画我猜”游戏。
发挥创意吧！在我们的 Discord 中分享你的作品，我们会展示一些我们最喜欢的东西。🪄
更多精彩内容 请关注我的个人公众号 公众号（办公AI智能小助手）或者 我的个人博客 https://blog.qife122.com/
对网络安全、黑客技术感兴趣的朋友可以关注我的安全公众号（网络安全技术点滴分享）
