Stable Diffusion WebUI

🚀 图像生成 MCP 服务器

本项目是一个 MCP 服务器，它提供基于 Stable Diffusion WebUI API（ForgeUI/AUTOMATIC-1111）的文字转图像生成功能，能帮助用户轻松实现文字到图像的转换。

🚀 快速开始

本服务器可提供基于 Stable Diffusion WebUI API 的文字转图像生成功能。在使用前，请确保满足以下先决条件，并完成相应的设置。

📦 安装指南

先决条件

• Node.js
• 访问带有 API 开启的 Stable Diffusion WebUI 实例
• WebUI 启动时必须启用 --api 标志

设置

1. 克隆仓库：

git clone https://github.com/Ichigo3766/image-gen-mcp.git
cd image-gen-mcp

2. 安装依赖项：

npm install

3. 构建服务器：

npm run build

4. 将服务器配置添加到您的环境：

{
  "mcpServers": {
    "image-gen": {
      "command": "node",
      "args": [
        "/path/to/image-gen-mcp/build/index.js"
      ],
      "env": {
        "NODE_ENV": "production",
        "PORT": "3000",
        "SD_WEBUI_URL": "http://localhost:7860",
        "AUTH_ENABLED": "false",
        "OUTPUT_DIR": "/path/to/output",
        "UPSCALER_1": "RealESRGAN_x2plus",
        "UPSCALER_2": "RealSR_x4plus"
      }
    }
  }
}

✨ 主要特性

工具

生成图像

使用文字转图像模型生成高质量图像。

获取 SD 模型

获取当前可用的 Stable Diffusion 模型列表。

设置 SD 模型

设置默认的 Stable Diffusion 模型。

上调分辨率

使用 Stable Diffusion 对图像进行分辨率提升。

参数

生成图像

• prompt (必填)：图像描述的文字提示。
• negative_prompt：不希望出现的内容提示。
• width (可选，默认 512)：输出图像的宽度。
• height (可选，默认 512)：输出图像的高度。
• samples (可选，默认 1)：生成图像的数量。
• num_inference_steps (可选，默认 20)：推理步骤数量。

获取 SD 模型

无参数，返回所有可用模型的列表。

设置 SD 模型

• model_id (必填)：要设置的模型 ID。

上调分辨率

• images (必填)：需要上调的图像文件路径数组。
• resize_mode (可选，默认来自环境变量)：0 表示倍数模式，1 表示维度模式。
• upscaling_resize (可选，默认来自环境变量)：倍数模式下的缩放比例。
• upscaling_resize_w (可选，默认来自环境变量)：维度模式下的目标宽度（以像素为单位）。
• upscaling_resize_h (可选，默认来自环境变量)：维度模式下的目标高度（以像素为单位）。
• upscaler_1 (可选，默认来自环境变量)：主上调器模型。
• upscaler_2 (可选，默认来自环境变量)：辅助上调器模型。
• output_path (可选，默认当前目录)：上调后图像的输出路径。

💻 使用示例

开发

使用自动重建进行开发：

npm run watch

📚 详细文档

错误处理

常见问题及解决方案：

1. 确保您的 Stable Diffusion WebUI 正在运行，并且启用了 --api 标志。
2. 检查 WebUI URL 是否可以从 MCP 服务器访问。
3. 如果使用了身份验证，请确保凭据正确。
4. 验证输出目录是否存在且具有写入权限。
5. 在上调分辨率时，确保输入图像文件存在且可读。

📄 许可证

此 MCP 服务器根据 MIT License 分发。这意味着您可以自由地使用、修改和分发软件，但需遵守 MIT License 的条款和条件。有关详细信息，请参阅项目存储库中的 LICENSE 文件。