MCP 服务器

Image Generation MCP Server

一个 MCP 服务器，它与 Stable Diffusion WebUI 集成，通过简单的 API 调用提供文本到图像的生成和图像放大功能。

Tools

generate_image

Generate an image using Stable Diffusion

get_sd_models

Get list of available Stable Diffusion models

set_sd_model

Set the active Stable Diffusion model

get_sd_upscalers

Get list of available upscaler models

upscale_images

Upscale one or more images using Stable Diffusion

README

image-gen MCP 服务器

一个使用 Stable Diffusion WebUI API (ForgeUI/AUTOMATIC-1111) 提供文本到图像生成功能的 MCP 服务器。

安装

前提条件

Node.js
能够访问已启用 API 的 Stable Diffusion WebUI 实例
启动 WebUI 时必须启用 --api 标志

设置

克隆仓库：

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

安装依赖：

npm install

构建服务器：

npm run build

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

{
  "mcpServers": {
    "image-gen": {
      "command": "node",
      "args": [
        "/path/to/image-gen-mcp/build/index.js"
      ],
      "env": {
        "SD_WEBUI_URL": "http://your-sd-webui-url:7860",
        "SD_AUTH_USER": "your-username",  // 可选：如果启用了身份验证
        "SD_AUTH_PASS": "your-password",  // 可选：如果启用了身份验证
        "SD_OUTPUT_DIR": "/path/to/output/directory",
        "SD_RESIZE_MODE": "0",           // 可选：放大模式 (0=倍数, 1=尺寸)
        "SD_UPSCALE_MULTIPLIER": "4",    // 可选：默认放大倍数
        "SD_UPSCALE_WIDTH": "512",       // 可选：默认放大宽度
        "SD_UPSCALE_HEIGHT": "512",      // 可选：默认放大高度
        "SD_UPSCALER_1": "R-ESRGAN 4x+", // 可选：默认主要放大器
        "SD_UPSCALER_2": "None"          // 可选：默认次要放大器
      }
    }
  }
}

使用您的值替换环境变量：

SD_WEBUI_URL: 您的 Stable Diffusion WebUI 实例的 URL
SD_AUTH_USER: 基本身份验证的用户名（如果已启用）
SD_AUTH_PASS: 基本身份验证的密码（如果已启用）
SD_OUTPUT_DIR: 将保存生成的图像的目录
SD_RESIZE_MODE: 默认放大模式（0 表示倍数，1 表示尺寸）
SD_UPSCALE_MULTIPLIER: 当 resize_mode 为 0 时的默认放大倍数
SD_UPSCALE_WIDTH: 当 resize_mode 为 1 时的默认目标宽度
SD_UPSCALE_HEIGHT: 当 resize_mode 为 1 时的默认目标高度
SD_UPSCALER_1: 默认主要放大器模型
SD_UPSCALER_2: 默认次要放大器模型

功能

工具

generate_image - 使用 Stable Diffusion 生成图像
- 参数：
  - prompt (必需): 所需图像的文本描述
  - negative_prompt: 要从图像中排除的内容
  - steps: 采样步数（默认值：4，范围：1-150）
  - width: 图像宽度（默认值：1024，范围：512-2048）
  - height: 图像高度（默认值：1024，范围：512-2048）
  - cfg_scale: CFG 比例（默认值：1，范围：1-30）
  - sampler_name: 采样算法（默认值：“Euler”）
  - scheduler_name: 调度算法（默认值：“Simple”）
  - seed: 随机种子（-1 表示随机）
  - batch_size: 要生成的图像数量（默认值：1，最大值：4）
  - restore_faces: 启用面部修复
  - tiling: 生成可平铺的图像
  - output_path: 生成图像的自定义输出路径
get_sd_models - 获取可用 Stable Diffusion 模型的列表
- 无需参数
- 返回模型名称数组
set_sd_model - 设置活动的 Stable Diffusion 模型
- 参数：
  - model_name (必需): 要设置为活动模型的名称
get_sd_upscalers - 获取可用放大器模型的列表
- 无需参数
- 返回放大器名称数组
upscale_images - 使用 Stable Diffusion 放大一个或多个图像
- 参数：
  - images (必需): 要放大的图像文件路径数组
  - resize_mode: 0 表示倍数模式，1 表示尺寸模式（默认值：来自 env）
  - upscaling_resize: 当 resize_mode=0 时的放大倍数（默认值：来自 env）
  - upscaling_resize_w: 当 resize_mode=1 时的目标宽度（以像素为单位）（默认值：来自 env）
  - upscaling_resize_h: 当 resize_mode=1 时的目标高度（以像素为单位）（默认值：来自 env）
  - upscaler_1: 主要放大器模型（默认值：来自 env）
  - upscaler_2: 次要放大器模型（默认值：来自 env）
  - output_path: 放大图像的自定义输出目录