← Back to home@Aenotr

local-imagegen-skill

DSH Skill: 用本机 GPU 上的 ComfyUI + Z-Image Turbo 离线生图（零 API 费用），含显卡能力自检、文生图与局部重绘 CLI

Stars
2
Language
Python
Created
Oct 4, 2026
Updated
Oct 5, 2026
Skills
GitHub repo

Introduction

local-imagegen

一个 DSH Skill：用本机 GPU 上的 ComfyUI + Z-Image Turbo 离线生成图片 —— 零 API 费用、无限次试错，也支持对已生成图片做局部重绘（inpainting）。

这个仓库只有前端（技能说明 + 脚本）。出图能力来自你自己的 ComfyUI 后端与本地模型。

特性

  • 零边际成本：本地推理，不消耗任何云端积分
  • 能力自检：出图前先探测显卡显存、驱动、CUDA、ComfyUI 安装与模型完整性，并强制向用户确认后才启用重负载 GPU 任务
  • 文生图 CLI：常用比例预设、自定义尺寸、固定 seed 复现、批量出图挑选
  • 局部重绘 CLI：矩形或阶梯形多边形蒙版，只改一小块而不整张重生成
  • 可搬运：脚本内部没有写死的用户路径，LOCAL_IMAGEGEN_ROOT 等环境变量即可换机器

环境要求

项目要求
显卡NVIDIA，显存 ≥ 6 GB（推荐 ≥ 8 GB）
驱动支持 CUDA 12.8（实测 577.05 可用；580+ 可上 cu130 加速内核）
Python3.10+（脚本只用标准库；inpaint.py 另需 Pillow）
ComfyUI后端服务，默认路径 D:\dsh\ComfyUI，可用环境变量覆盖
模型Z-Image Turbo int8 + Qwen3-4B 文本编码器 + VAE，合计约 11.3 GB

后端怎么搭见 local-imagegen/SETUP-comfyui.md。

安装

  1. 把 local-imagegen/ 整个目录放进 DSH 的技能根目录（默认 ~/.agents/skills/）：

    $dest = "$env:USERPROFILE\.agents\skills"
New-Item -ItemType Directory -Force -Path $dest | Out-Null
Copy-Item -Path .\local-imagegen -Destination $dest -Recurse -Force

  2. 确认技能加载器已启用（最常见的坑）：DSH 需要启用本地技能提供方 @deepseek-ai/dsh-skill-filesystem，否则技能目录永远是空的，agent 报 skill "local-imagegen" is unknown or no longer available —— 文件都在，只是没人来读。 完整排查步骤见 local-imagegen/INSTALL.md 的 2.1 节。

  3. 验证：

    python local-imagegen\scripts\check_capability.py --json

使用

技能注册后，直接对 agent 说"用 local-imagegen 画……"即可。也可以直接调脚本：

# 提示词写进 UTF-8 文件再传（Windows 命令行直传中文会乱码）
python local-imagegen\scripts\generate.py --prompt-file prompt.txt --ratio 16:9
python local-imagegen\scripts\generate.py --prompt-file prompt.txt --width 1344 --height 768 --seed 12345

# 局部重绘：蒙版越贴合目标物体越好
python local-imagegen\scripts\inpaint.py --image out.png --poly "1102,200 1126,200 1126,178 1246,178 1246,262 1126,262 1126,243" --prompt-file fix.txt

环境变量（都有默认值，换机器时覆盖即可）：

变量默认值说明
LOCAL_IMAGEGEN_ROOTD:\dsh\ComfyUIComfyUI 安装目录
LOCAL_IMAGEGEN_HOSThttp://127.0.0.1:8188服务地址
LOCAL_IMAGEGEN_UNET / _CLIP / _VAE见 INSTALL.md模型文件名

已知边界（请勿过度期待）

  • CFG=1 让负面提示词数学上失效：Z-Image Turbo 是蒸馏模型，不要写"不要 XX"，只做正向描述。
  • 细粒度机械结构不可靠，且失败是静默的：类别正确性高（是坦克），个体正确性低（是哪一型坦克）。
  • 不适合：技术插图、真实人物/地点还原、画面内文字、精确数量与空间关系。
  • 需要事实准确性时，正路是参考图 + 图生图/ControlNet，或挂对应 LoRA。

目录结构

local-imagegen/
├── SKILL.md                     技能说明：执行流程、边界、环境坑（agent 读这个）
├── INSTALL.md                   安装与排查
├── SETUP-comfyui.md             后端搭建指南
└── scripts/
    ├── check_capability.py      出图前的硬件能力检测（必须先跑）
    ├── ensure_server.ps1        探测/拉起 ComfyUI 服务
    ├── comfy_client.py          ComfyUI 客户端 + 工作流构建
    ├── generate.py              文生图 CLI
    ├── inpaint.py               局部重绘 CLI（--denoise / --mask-blur 可控强度与软边）
    └── hybrid_patch.py          跨阶段像素级补丁：裁剪 → 重绘 → 原分辨率贴回，内置 --align-from 自动重定位

附带的 DSH 插件：dsh-image-modes

本仓库还包含一个把三种出图方式整合成一个工具的 DSH 插件，代码在 dsh-image-modes/：

gen_image(mode = "cloud" | "local" | "hybrid", prompt, ...)   # 一个入口三种模式
image_modes_status()                                          # 开工前自检：三种方式各自能不能用
understand_image(images, prompt)                              # 识图：读图中文字、多图对比、看图验收
  • cloud：远端 OpenAI 兼容图像 API，最高 4K 档（2K ≈ 3.9MP），看得懂画面，但没有蒙版、没有 seed，每次都整幅重绘
  • local：本机 ComfyUI + Z-Image Turbo，12–32 秒/张、零成本、seed 可复现、像素级蒙版重绘，上限约 1.03MP
  • hybrid：本地先出草稿或打底，再交云端高清重构 —— 实测唯一同时保住"完成度 + 参考角色神态"的路线

插件把实测出来的取舍固化进了代码：mode 参数没有默认值（强制显式选择），工具描述里写明"调用前必须先问用户并说明取舍"，全身立绘提醒用 9:16、本地重绘提醒蒙版要贴合物体，云端带 4 次重试退避，ComfyUI 掉线可自动拉起。

安装

推荐：让 agent 帮你装。 对 DSH 里的 agent 说一句"把仓库里的 dsh-image-modes 装进 profile"，它会复制到 <profile>/vendor/、改 package.json、跑 pnpm install。

手动安装：

  1. 把 dsh-image-modes 整个目录复制到 <profile>/vendor/dsh-image-modes
  2. 在 <profile>/package.json 里：
{
  "dependencies": { "dsh-image-modes": "file:./vendor/dsh-image-modes" },
  "dsh": { "profile": { "bundles": ["@deepseek-ai/dsh-base", "@deepseek-ai/dsh-web-app", "dsh-image-modes"] } }
}
  1. 在 profile 目录执行 pnpm install，然后重载 profile

安装后三个工具会出现在 agent 的工具表里。云端密钥不需要重复配置：插件复用 IMAGE_API_KEY / IMAGE_API_BASE 这套凭据引用。详见 dsh-image-modes/README.md。

许可

MIT

更新记录

  • 1.2.1 —— ensure_server.ps1 修两处：ComfyUI 静默崩溃时不再无据可查（输出重定向到 <root>/logs/comfyui.out|err.log）；脚本改为纯 ASCII + 参数表，修掉"中文注释在 GBK 主机上破坏 PowerShell 行尾续行"的语法崩。另在本仓库加入 dsh-image-modes 插件。
  • 1.2.0 —— 局部重绘可控化：inpaint.py 新增 --denoise（0.5–0.7 保住轮廓改外观、0.85–1.0 去物体填空白）与 --mask-blur（喂给模型的软边蒙版，消除接缝环）；新增 hybrid_patch.py 负责"云端出高清 → 本地精确修"的裁剪-重绘-贴回，并能自动换算跨阶段的坐标漂移；SKILL.md 补上"蒙版必须贴合目标物体"与"跨阶段必须先配准"两条实测教训。
  • 1.1.0 —— 可移植化：去掉写死的用户路径，改用技能基目录相对路径；补上安装排查（技能加载器必须已启用）与分发必备条件说明。
  • 1.0.0 —— 首版：能力自检、文生图与局部重绘 CLI。

致谢

Z-Image Turbo（扩散模型）、Qwen3-4B（文本编码器）、ComfyUI。