koishi-plugin-p-draw 1.2.6 → 1.2.8

package/index.js

@@ -191,7 +191,16 @@ exports.Config = Schema.object({
   pollInterval: Schema.number().default(2).description('生成状态查询间隔（秒）'),
 
   // 模型文件
-  unetName: Schema.string().default('anima_baseV10.safetensors').description('主模型文件名'),
+  unetName: Schema.string().default('anima-base-v1.0.safetensors').description('主模型文件名（需与 ComfyUI models/diffusion_models 下的文件名完全一致，含连字符；默认 anima-base-v1.0.safetensors）'),
+  i2iUnetName: Schema.string().default('anima-base-v1.0.safetensors').description('「换风格（漫画化）」工作流专用主模型文件名。LLLite 权重按 block 数逐块训练，当前 LLLite 权重为 28-block，必须搭配 28-block 模型（anima-base-v1.0 / anima-aesthetic-v1.1）；不要用 40-block 的 Anima-2.9B，否则报 depth_embed slices missing。留空则用主模型 unetName'),
+  modelParams: Schema.dict(Schema.object({
+    samplerName: Schema.string().description('采样器（如 er_sde / res_2s / dpmpp_2m）'),
+    scheduler: Schema.string().description('调度器（如 simple / beta57 / normal）'),
+    steps: Schema.number().description('采样步数'),
+    cfg: Schema.number().description('CFG 强度'),
+    width: Schema.number().description('默认宽度'),
+    height: Schema.number().description('默认高度'),
+  }).description('该模型的生成参数覆盖')).default({}).description('每个模型独立参数覆盖：key=模型文件名（如 anima-base-v1.0.safetensors、Anima-2.9B-preview-v1.safetensors、anima-aesthetic-v1.1.safetensors），value=要覆盖的参数（samplerName/scheduler/steps/cfg/width/height）。只填需要覆盖的字段，不填的用全局默认。匹配不区分大小写与 -/_。'),
   clipName: Schema.string().default('qwen_3_06b_base.safetensors').description('文本编码器文件名'),
   vaeName: Schema.string().default('qwen_image_vae.safetensors').description('VAE 文件名'),
 
@@ -566,22 +575,28 @@ function animaI2IWorkflow(cfg, prompt, negativePrompt, width, height, steps, cfg
   }
 }
 
-// 「换风格（漫画化）」i2i 工作流：LoadImage → Canny → AnimaLLLiteApply_sdscripts 锁构图，
+// 「换风格（漫画化）」i2i 工作流：LoadImage → 预处理器（LineArt / Canny）→ AnimaLLLiteApply_sdscripts 锁构图，
 // VAEEncode 原图作起点，KSampler 用较高 denoise 整体转二次元风格。
 // 注意：Anima 是 MiniTrainDIT 架构（隐藏层 3584 维），与 Qwen-Image 系 ControlNet 不兼容
 // （InstantX 控制网期望 3584 维文本嵌入，Anima 只给 1024 维 → LayerNorm 崩溃）。
 // 因此这里使用 Anima 原生支持的 ControlNet-LLLite（kohya-ss/ComfyUI-Anima-LLLite），
 // 节点直接对 MODEL 打补丁并返回补丁后的模型，交给 KSampler。
-function animaStyleI2IWorkflow(cfg, prompt, negativePrompt, width, height, steps, cfgVal, seed, inputImage, controlNetModel, controlNetStrength, denoise) {
+// 预处理器：comfyui_controlnet_aux 的 AnimeLineArt/LineArt 节点（需安装该自定义节点）；
+// 未安装时回退 ComfyUI 内置 Canny。anima-lllite-lineart 权重是按 lineart 训练的，用 LineArt 比 Canny 更贴。
+function animaStyleI2IWorkflow(cfg, prompt, negativePrompt, width, height, steps, cfgVal, seed, inputImage, controlNetModel, controlNetStrength, denoise, preprocessor) {
+  const detectRes = Math.max(width, height)
+  const preprocessorNode = preprocessor === 'AnimeLineArtPreprocessor' || preprocessor === 'LineArtPreprocessor'
+    ? { class_type: preprocessor, inputs: { image: ['13', 0], detect_resolution: detectRes, resolution: detectRes } }
+    : { class_type: 'Canny', inputs: { image: ['13', 0], low_threshold: 0.4, high_threshold: 0.8 } }
   return {
-    '44': { class_type: 'UNETLoader', inputs: { unet_name: cfg.unetName, weight_dtype: 'fp8_e4m3fn' } },
+    '44': { class_type: 'UNETLoader', inputs: { unet_name: cfg.i2iUnetName || cfg.unetName, weight_dtype: 'fp8_e4m3fn' } },
     '45': { class_type: 'CLIPLoader', inputs: { clip_name: cfg.clipName, type: 'stable_diffusion', device: 'default' } },
     '15': { class_type: 'VAELoader', inputs: { vae_name: cfg.vaeName } },
     '13': { class_type: 'LoadImage', inputs: { image: inputImage } },
     '30': { class_type: 'VAEEncode', inputs: { pixels: ['13', 0], vae: ['15', 0] } },
     '11': { class_type: 'CLIPTextEncode', inputs: { text: prompt, clip: ['45', 0] } },
     '12': { class_type: 'CLIPTextEncode', inputs: { text: negativePrompt, clip: ['45', 0] } },
-    '70': { class_type: 'Canny', inputs: { image: ['13', 0], low_threshold: 0.4, high_threshold: 0.8 } },
+    '70': preprocessorNode,
     '71': { class_type: 'AnimaLLLiteApply_sdscripts', inputs: { model: ['44', 0], lllite_name: controlNetModel, image: ['70', 0], strength: controlNetStrength, start_percent: 0, end_percent: 1, preserve_wrapper: true } },
     '19': {
       class_type: 'KSampler',
@@ -649,7 +664,7 @@ function buildI2IWorkflow(cfg, prompt, negativePrompt, width, height, steps, cfg
       return {
         kind: 'controlnet',
         denoise,
-        promptBody: animaStyleI2IWorkflow(cfg, prompt, negativePrompt, width, height, steps, cfgVal, seed, inputImage, caps.controlNet.model, Number(cfg.i2iControlNetStrength) || 0.7, denoise),
+        promptBody: animaStyleI2IWorkflow(cfg, prompt, negativePrompt, width, height, steps, cfgVal, seed, inputImage, caps.controlNet.model, Number(cfg.i2iControlNetStrength) || 0.7, denoise, caps.controlNet.preprocessor),
       }
     }
     // 没有 ControlNet：换风格又不想崩构图，denoise 自动压到 0.5 以内
@@ -1431,10 +1446,12 @@ exports.apply = async function apply(ctx, cfg) {
       const vaeList = availableModels(objectInfo, 'VAELoader', 'vae_name')
       payload.unet_available = unetList.includes(cfg.unetName)
       payload.unet_models = unetList
+      payload.i2i_unet_available = unetList.includes(String(cfg.i2iUnetName || '').trim() || cfg.unetName)
       payload.clip_available = clipList.includes(cfg.clipName)
       payload.vae_available = vaeList.includes(cfg.vaeName)
     } else {
       payload.unet_available = undefined
+      payload.i2i_unet_available = undefined
       payload.clip_available = undefined
       payload.vae_available = undefined
     }
@@ -1480,6 +1497,7 @@ exports.apply = async function apply(ctx, cfg) {
       `版本：${payload.comfyui_version || '未知'}`,
       `GPU：${payload.gpu || '未知'}（显存 ${payload.vram_total_mb}MB / 空闲 ${payload.vram_free_mb}MB）`,
       `主模型：${cfg.unetName} ${modelStatus(cfg.unetName, payload.unet_available)}`,
+      `换风格模型：${cfg.i2iUnetName || cfg.unetName} ${modelStatus(cfg.i2iUnetName || cfg.unetName, payload.i2i_unet_available)}`,
       `文本编码器：${cfg.clipName} ${modelStatus(cfg.clipName, payload.clip_available)}`,
       `VAE：${cfg.vaeName} ${modelStatus(cfg.vaeName, payload.vae_available)}`,
       `可用尺寸：${payload.allowed_sizes.join('、')}`,
@@ -1590,6 +1608,26 @@ exports.apply = async function apply(ctx, cfg) {
     return String(chosen).trim()
   }
 
+  // 读取某模型在 modelParams 中的独立覆盖参数（只允许采样相关字段，避免误改 unet/clip/vae 等路径）
+  const MODEL_OVERRIDE_KEYS = ['samplerName', 'scheduler', 'steps', 'cfg', 'width', 'height']
+  function resolveModelParams(name) {
+    if (!name || !cfg.modelParams || typeof cfg.modelParams !== 'object') return {}
+    const norm = (s) => String(s || '').trim().toLowerCase().replace(/[-_]/g, '~')
+    const key = norm(name)
+    if (!key) return {}
+    const normalized = {}
+    for (const [k, v] of Object.entries(cfg.modelParams)) {
+      normalized[norm(k)] = v
+    }
+    const raw = normalized[key]
+    if (!raw || typeof raw !== 'object') return {}
+    const out = {}
+    for (const k of MODEL_OVERRIDE_KEYS) {
+      if (raw[k] !== undefined && raw[k] !== null && raw[k] !== '') out[k] = raw[k]
+    }
+    return out
+  }
+
   // 名称匹配：精确 > 前缀唯一 > 包含唯一；多个匹配返回 { multiple: [...] }
   function matchUnetModel(input, list) {
     // 规范化：连字符与下划线视为等价（用户常写 anima-aesthetic，模型名是 anima_aesthetic）
@@ -1611,15 +1649,16 @@ exports.apply = async function apply(ctx, cfg) {
     const sizes = parseAllowedSizes()
     const requestedWidth = (size && size[0]) || overrides.width || cfg.width
     const requestedHeight = (size && size[1]) || overrides.height || cfg.height
-    const width = Math.round(Number(requestedWidth) || cfg.width)
-    const height = Math.round(Number(requestedHeight) || cfg.height)
-    const steps = Math.round(Number(overrides.steps) || cfg.steps)
-    const cfgVal = Number(overrides.cfg) || cfg.cfg
+    const unetName = overrides.unet || cfg.unetName
+    // 应用该模型的独立参数覆盖（modelParams），再叠加命令级 overrides（overrides.steps/cfg 优先于模型级）
+    const modelSpecific = resolveModelParams(unetName)
+    const workCfg = Object.assign({}, cfg, modelSpecific, { unetName })
+    const width = Math.round(Number(requestedWidth) || workCfg.width)
+    const height = Math.round(Number(requestedHeight) || workCfg.height)
+    const steps = Math.round(Number(overrides.steps) || workCfg.steps)
+    const cfgVal = Number(overrides.cfg) || workCfg.cfg
     const seed = Number(overrides.seed) || crypto.randomInt(1, 2 ** 32 - 1)
     const negativePrompt = overrides.negativePrompt || cfg.negativePrompt || ''
-    const unetName = overrides.unet || cfg.unetName
-
-    const workCfg = unetName && unetName !== cfg.unetName ? Object.assign({}, cfg, { unetName }) : cfg
     const i2iImage = overrides.i2iImage
     if (i2iImage && cfg.customWorkflowEnabled && cfg.customWorkflowPath) {
       return { ok: false, message: 'i2i（以图生图）暂不支持自定义工作流（customWorkflowEnabled），请关闭后再试。' }
@@ -3187,11 +3226,12 @@ exports.apply = async function apply(ctx, cfg) {
     return data.name
   }
 
-  // 检测 i2i 可用能力：Anima ControlNet-LLLite 节点与权重（comfyui-anima-lllite / kohya-ss/ComfyUI-Anima-LLLite）
-  // 与 Anima IP-Adapter 专用节点（comfyui-anima-ipadapter）。检测失败全部视为不可用（回退普通 img2img）。
+  // 检测 i2i 可用能力：Anima ControlNet-LLLite 节点与权重（comfyui-anima-lllite / kohya-ss/ComfyUI-Anima-LLLite）、
+  // LineArt 预处理器（comfyui_controlnet_aux，可选）与 Anima IP-Adapter 专用节点（comfyui-anima-ipadapter）。
+  // 检测失败全部视为不可用（回退普通 img2img）。
   // 注意：Qwen-Image InstantX ControlNet 与 Anima（MiniTrainDIT，3584 维）架构不兼容，这里只认 LLLite 权重。
   async function detectI2ICapabilities() {
-    const caps = { controlNet: { available: false, model: null }, ipAdapter: { available: false } }
+    const caps = { controlNet: { available: false, model: null, preprocessor: 'Canny' }, ipAdapter: { available: false } }
     let info
     try {
       info = await getObjectInfoCached()
@@ -3199,9 +3239,16 @@ exports.apply = async function apply(ctx, cfg) {
       return caps
     }
     if (!info || typeof info !== 'object') return caps
+    // LineArt 预处理器（comfyui_controlnet_aux）：AnimeLineArt 更贴近漫画线条，优先；LineArt 次之；都没有则回退 Canny
+    if (info.AnimeLineArtPreprocessor) {
+      caps.controlNet.preprocessor = 'AnimeLineArtPreprocessor'
+    } else if (info.LineArtPreprocessor) {
+      caps.controlNet.preprocessor = 'LineArtPreprocessor'
+    }
     // Anima ControlNet-LLLite：lllite_name 从 models/controlnet 目录读取，只挑 anima-lllite 系权重
     if (info.AnimaLLLiteApply_sdscripts) {
       const llliteModels = availableModels(info, 'AnimaLLLiteApply_sdscripts', 'lllite_name')
+      if (cfg.outputLogs) logger.info(`[p-draw] i2i LLLite 节点存在，models/controlnet 文件列表=${JSON.stringify(llliteModels)}`)
       if (llliteModels.length) {
         const configured = String(cfg.controlNetModel || '').trim()
         const norm = (s) => String(s).toLowerCase().replace(/[-_ ]/g, '')
@@ -3213,6 +3260,15 @@ exports.apply = async function apply(ctx, cfg) {
         caps.controlNet.model = configured && llliteModels.includes(configured) ? configured : best
         caps.controlNet.available = !!caps.controlNet.model
       }
+      // 换风格专用主模型必须是 28-block（与 LLLite 权重匹配），检测它是否已安装
+      const styleModel = String(cfg.i2iUnetName || '').trim() || cfg.unetName
+      const unetList = availableModels(info, 'UNETLoader', 'unet_name')
+      if (unetList.length && !unetList.includes(styleModel)) {
+        logger.warn(`[p-draw] i2i 换风格专用主模型 ${styleModel} 不在 models/diffusion_models 中（现有：${JSON.stringify(unetList)}）。LLLite 权重为 28-block，请安装 anima-base-v1.0 或修改 i2iUnetName 配置，否则换风格会报 depth_embed slices missing`)
+        caps.controlNet.modelMismatch = true
+      }
+    } else if (cfg.outputLogs) {
+      logger.info('[p-draw] i2i 未检测到 AnimaLLLiteApply_sdscripts 节点（请确认已安装 kohya-ss/ComfyUI-Anima-LLLite 并重启 ComfyUI）')
     }
     if (info.AnimaIPAdapterLoader && info.AnimaIPAdapterApply && info.AnimaSiglipeEncodeImage) {
       caps.ipAdapter.available = true

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-p-draw",
   "description": "p点-绘图，连接本地 ComfyUI 生图。支持自然语言优化、尺寸选择、画师组、固定角色、生成队列，需要配合 p-qiandao 使用",
-  "version": "1.2.6",
+  "version": "1.2.8",
   "main": "index.js",
   "files": [
     "index.js",

package/readme.md

@@ -67,6 +67,8 @@ plugins:
 |----|------|
 | `comfyuiBaseUrl` | ComfyUI API 地址（默认 `http://127.0.0.1:8188`） |
 | `unetName` / `clipName` / `vaeName` | 模型文件名，需与 ComfyUI 下拉框中的文件名一致 |
+| `i2iUnetName` | 「换风格（漫画化）」专用主模型文件名，默认 `anima-base-v1.0.safetensors`。LLLite 权重按 block 数逐块训练（当前权重为 28-block），必须配 28-block 模型（`anima-base-v1.0` / `anima-aesthetic-v1.1`）；**不要用 40-block 的 `Anima-2.9B`**，否则报 `depth_embed slices missing`。留空则用 `unetName` |
+| `modelParams` | 每个模型独立参数覆盖（字典）：`key`=模型文件名，`value`=要覆盖的 `samplerName` / `scheduler` / `steps` / `cfg` / `width` / `height`。让不同模型用各自最佳采样参数，只填需要覆盖的字段，命令级 `--steps`/`--cfg` 仍优先 |
 | `customWorkflowEnabled` / `customWorkflowPath` | 使用自定义工作流 JSON（相对插件目录） |
 | `width` / `height` / `allowedSizes` | 默认尺寸与可用尺寸列表 |
 | `steps` / `cfg` / `samplerName` / `scheduler` | 采样参数 |
@@ -111,5 +113,5 @@ plugins:
   - 券为一次性、**按张数消耗**：`x3` 需 3 张，每张图独立做一次 LLM 优化；券不足时该批不优化也不扣券（提示 `token-short`）。等待回复时间由 `couponAskTimeout`（秒）控制。
   - **管理员在全局关闭时自动免费优化**（不耗券）；`无优化` 原样模式不消耗、不询问。平台不支持交互确认时退回自动消耗逻辑。
 - **未使用优化的提示**：单人生图没走 LLM 优化时会明确提示原因——未配置 LLM、未消耗优化券、或 `无优化` 原样模式；多人指令强依赖 LLM 规划，未配置 `llmBaseUrl`/`llmModel` 时会直接提示「多人指令需要 LLM 规划」。
-- **i2i 增强模式依赖的节点**：换风格（漫画化）需要 ComfyUI 安装 [kohya-ss/ComfyUI-Anima-LLLite](https://github.com/kohya-ss/ComfyUI-Anima-LLLite) 自定义节点，并在 `models/controlnet/` 放一个 `anima-lllite` 系权重（`AnimaLLLiteApply_sdscripts` 节点下拉可见）。**注意：Anima 是 MiniTrainDIT 架构（3584 维），与 Qwen-Image 系 ControlNet（如 `Qwen-Image-InstantX-ControlNet-Union`）不兼容，装了也不会被本插件使用。** 换装换姿势需要安装 [comfyui-anima-ipadapter](https://github.com/Wenaka2004/comfyui-anima-ipadapter) 自定义节点并配置 `i2iIPAdapterPath`。两者缺失时都会自动回退普通 img2img 并提示，不影响出图。
+- **i2i 增强模式依赖的节点**：换风格（漫画化）需要 ComfyUI 安装 [kohya-ss/ComfyUI-Anima-LLLite](https://github.com/kohya-ss/ComfyUI-Anima-LLLite) 自定义节点，并在 `models/controlnet/` 放一个 `anima-lllite` 系权重（`AnimaLLLiteApply_sdscripts` 节点下拉可见）。**LLLite 权重按 block 数逐块训练，必须与所用模型一致：当前权重为 28-block，换风格工作流默认用 `i2iUnetName`（`anima-base-v1.0`，28-block）；若用 40-block 的 `Anima-2.9B` 会报 `depth_embed slices missing for module idx(es) [28..39]`。****注意：Anima 是 MiniTrainDIT 架构（3584 维），与 Qwen-Image 系 ControlNet（如 `Qwen-Image-InstantX-ControlNet-Union`）不兼容，装了也不会被本插件使用。** 可选安装 [comfyui_controlnet_aux](https://github.com/Fannovel16/comfyui_controlnet_aux) 获得 LineArt 预处理器（优先 `AnimeLineArtPreprocessor`，其次 `LineArtPreprocessor`），线条比内置 Canny 更贴合 lllite-lineart 权重；未安装时自动回退 Canny。换装换姿势需要安装 [comfyui-anima-ipadapter](https://github.com/Wenaka2004/comfyui-anima-ipadapter) 自定义节点并配置 `i2iIPAdapterPath`。以上节点缺失时都会自动回退普通 img2img 并提示，不影响出图。
 - 本地模型不一定认识新角色，可用 `添加角色` 手动补充角色 tags。