图像生成 MCP 服务器

一个提供基于 Stable Diffusion WebUI API（ForgeUI/AUTOMATIC-1111）的文字转图像生成功能的 MCP 服务器。

安装

先决条件

• 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 文件。