一个中二的 AI 绘图机器人。

仅限 Discord。

机器人支持私聊（DM）和群聊触发。

注意：机器人的交互在群聊中对所有人可见。

由于 Discord 的 App Command 不支持携带附件，图片转图片需要手工输入/repaint命令激活功能。

在参数不足的情况下机器人会通过单独的UI提示你输入更多的参数。

在群聊中，需要@机器人 /repaint才能激活命令。

graph
    A[sd_work_manager] --- B[sd_work_node]
    A[sd_work_manager] --- C[sd_work_node]
    A[sd_work_manager] --- D[sd_work_node]
    E[ai_painting_bot] --- A[sd_work_manager]
    F[discord] --- E[ai_painting_bot]
    G[users] --- F[discord]

• 依赖服务
  • sd_work_node
    • 不低于 6G 显存的 GPU 服务器
    • 在 Tesla P4（7.5G）环境下部署成功，可以工作在中等内存优化级别下
  • sd_work_manager
• Python >= 3.6
• 相关模型数据（Checkpoint、Embedding、Hypernetwork），具体见sd_work_node配置

python3 -m venv venv
source venv/bin/activate
pip install -r requirements.txt

• config.json

{
  "bot_token": "DISCORD BOT TOKEN HERE",
  "sd_api_base_url": "https://localhost",
  "sd_api_prefix": "/api",
  "sd_api_secret": "WEB API SECRET HERE",
  "available_modules": [
    "anime",
    "furry_kemono"
  ]
}

• https_proxy

当环境变量中配置有该值时，会被用于作为到 discord 的代理地址。

source venv/bin/activate
python3 ./main.py --config ./config.json

MIT License.