image-gen MCP Server

image-gen 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": {
        "SD_WEBUI_URL": "http://your-sd-webui-url:7860",
        "SD_AUTH_USER": "your-username",  // Optional: if authentication is enabled
        "SD_AUTH_PASS": "your-password",  // Optional: if authentication is enabled
        "SD_OUTPUT_DIR": "/path/to/output/directory",
        "SD_RESIZE_MODE": "0",           // Optional: upscaling mode (0=multiplier, 1=dimensions)
        "SD_UPSCALE_MULTIPLIER": "4",    // Optional: default upscale multiplier
        "SD_UPSCALE_WIDTH": "512",       // Optional: default upscale width
        "SD_UPSCALE_HEIGHT": "512",      // Optional: default upscale height
        "SD_UPSCALER_1": "R-ESRGAN 4x+", // Optional: default primary upscaler
        "SD_UPSCALER_2": "None"          // Optional: default secondary upscaler
      }
    }
  }
}

将环境变量替换为你的值：

• 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 表示尺寸模式（默认值：取自环境变量）
    • upscaling_resize：resize_mode=0 时的放大倍率（默认值：取自环境变量）
    • upscaling_resize_w：resize_mode=1 时的目标宽度（像素）（默认值：取自环境变量）
    • upscaling_resize_h：resize_mode=1 时的目标高度（像素）（默认值：取自环境变量）
    • upscaler_1：主要放大器模型（默认值：取自环境变量）
    • upscaler_2：次要放大器模型（默认值：取自环境变量）
    • output_path：放大图像的自定义输出目录

开发

使用自动重新构建进行开发：

npm run watch

错误处理

常见问题及解决方法：

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

许可证

此 MCP 服务采用 MIT 许可证。您可以自由使用、修改和分发此软件，但须遵守 MIT 许可证的条款和条件。详情请参阅项目仓库中的 LICENSE 文件。