koishi-plugin-p-draw 1.2.7 → 1.2.9

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 文件名'),
 
@@ -580,7 +589,7 @@ function animaStyleI2IWorkflow(cfg, prompt, negativePrompt, width, height, steps
     ? { 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 } },
@@ -1437,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
     }
@@ -1486,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('、')}`,
@@ -1596,6 +1608,27 @@ 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] === '') continue
+      out[k] = typeof raw[k] === 'string' ? raw[k].trim() : raw[k]
+    }
+    return out
+  }
+
   // 名称匹配：精确 > 前缀唯一 > 包含唯一；多个匹配返回 { multiple: [...] }
   function matchUnetModel(input, list) {
     // 规范化：连字符与下划线视为等价（用户常写 anima-aesthetic，模型名是 anima_aesthetic）
@@ -1617,15 +1650,19 @@ 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 })
+    // 防御：sampler/scheduler 配置若带尾随空格会导致 ComfyUI 报 "Value not in list"，统一 trim
+    if (typeof workCfg.samplerName === 'string') workCfg.samplerName = workCfg.samplerName.trim()
+    if (typeof workCfg.scheduler === 'string') workCfg.scheduler = workCfg.scheduler.trim()
+    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），请关闭后再试。' }
@@ -3227,6 +3264,13 @@ 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）')
     }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-p-draw",
   "description": "p点-绘图，连接本地 ComfyUI 生图。支持自然语言优化、尺寸选择、画师组、固定角色、生成队列，需要配合 p-qiandao 使用",
-  "version": "1.2.7",
+  "version": "1.2.9",
   "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_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 并提示，不影响出图。
+- **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。