mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-06-25 12:31:15 -03:00
71a459422f
- Add genParamsMapper.js: sampler/scheduler display→internal mapping, combined-name parsing, widget matching - Add sendGenParamsToWorkflow() in uiHelpers.js: resolves sampler, fetches registry by send_gen_params marker, sends via update-node-widget - Add send-params-btn UI in showcase hover panel and recipe modal - Add flashWidget() in workflow_registry.js: text-color visual cue on updated widget values (Vue: inline style + CSS, canvas: property shadow) - Add silent option to sendWidgetValueToNodes for consolidated toast - Normalize param display labels (cfg_scale→CFG, etc.) in recipe modal - Add 33 tests for genParamsMapper; update existing test assertions
297 lines
11 KiB
JavaScript
297 lines
11 KiB
JavaScript
/**
|
|
* genParamsMapper.js
|
|
* Maps display/recipe generation parameter values (sampler, scheduler) to
|
|
* ComfyUI internal widget values, enabling "Send Gen Params to Workflow".
|
|
*
|
|
* Strategy (3 layers):
|
|
* 1. Direct lookup via SAMPLER_DISPLAY_TO_INTERNAL
|
|
* 2. Combined-name parsing (e.g. "Euler a Karras" → sampler + scheduler)
|
|
* 3. Graceful skip for model-specific / unrecognized values
|
|
*/
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// Sampler display name → internal name (ComfyUI KSampler.SAMPLERS / SAMPLER_NAMES)
|
|
// ---------------------------------------------------------------------------
|
|
const SAMPLER_DISPLAY_TO_INTERNAL = {
|
|
// --- Euler family ---
|
|
'Euler': 'euler',
|
|
'euler': 'euler',
|
|
'Euler a': 'euler_ancestral',
|
|
'Euler A': 'euler_ancestral',
|
|
'Euler ancestral': 'euler_ancestral',
|
|
'Euler Ancestral': 'euler_ancestral',
|
|
'euler_ancestral': 'euler_ancestral',
|
|
|
|
// --- Heun ---
|
|
'Heun': 'heun',
|
|
'heun': 'heun',
|
|
'Heun++': 'heunpp2',
|
|
'heunpp2': 'heunpp2',
|
|
|
|
// --- DPM2 ---
|
|
'DPM2': 'dpm_2',
|
|
'DPM 2': 'dpm_2',
|
|
'dpm_2': 'dpm_2',
|
|
'DPM2 a': 'dpm_2_ancestral',
|
|
'DPM2 Ancestral': 'dpm_2_ancestral',
|
|
'dpm_2_ancestral': 'dpm_2_ancestral',
|
|
|
|
// --- LMS ---
|
|
'LMS': 'lms',
|
|
'lms': 'lms',
|
|
|
|
// --- DPM fast / adaptive ---
|
|
'DPM fast': 'dpm_fast',
|
|
'DPM Fast': 'dpm_fast',
|
|
'dpm_fast': 'dpm_fast',
|
|
'DPM adaptive': 'dpm_adaptive',
|
|
'DPM Adaptive': 'dpm_adaptive',
|
|
'dpm_adaptive': 'dpm_adaptive',
|
|
|
|
// --- DPM++ 2S ancestral ---
|
|
'DPM++ 2S a': 'dpmpp_2s_ancestral',
|
|
'DPM++ 2S A': 'dpmpp_2s_ancestral',
|
|
'DPM++ 2S Ancestral': 'dpmpp_2s_ancestral',
|
|
'dpmpp_2s_ancestral': 'dpmpp_2s_ancestral',
|
|
|
|
// --- DPM++ SDE ---
|
|
'DPM++ SDE': 'dpmpp_sde',
|
|
'dpmpp_sde': 'dpmpp_sde',
|
|
|
|
// --- DPM++ 2M ---
|
|
'DPM++ 2M': 'dpmpp_2m',
|
|
'dpmpp_2m': 'dpmpp_2m',
|
|
|
|
// --- DPM++ 2M SDE ---
|
|
'DPM++ 2M SDE': 'dpmpp_2m_sde',
|
|
'dpmpp_2m_sde': 'dpmpp_2m_sde',
|
|
|
|
// --- DPM++ 3M SDE ---
|
|
'DPM++ 3M SDE': 'dpmpp_3m_sde',
|
|
'dpmpp_3m_sde': 'dpmpp_3m_sde',
|
|
|
|
// --- Others ---
|
|
'DDIM': 'ddim',
|
|
'ddim': 'ddim',
|
|
'DDPM': 'ddpm',
|
|
'ddpm': 'ddpm',
|
|
'LCM': 'lcm',
|
|
'lcm': 'lcm',
|
|
'IPNDM': 'ipndm',
|
|
'ipndm': 'ipndm',
|
|
'DEIS': 'deis',
|
|
'deis': 'deis',
|
|
'UniPC': 'uni_pc',
|
|
'unipc': 'uni_pc',
|
|
'uni_pc': 'uni_pc',
|
|
|
|
// --- Restart / res_multistep ---
|
|
'Restart': 'res_multistep',
|
|
'res_multistep': 'res_multistep',
|
|
|
|
// --- ER SDE ---
|
|
'ER SDE': 'er_sde',
|
|
'E-R SDE': 'er_sde',
|
|
'er_sde': 'er_sde',
|
|
|
|
// --- SA Solver ---
|
|
'SA Solver': 'sa_solver',
|
|
'SA solver': 'sa_solver',
|
|
'sa_solver': 'sa_solver',
|
|
|
|
// --- Seeds ---
|
|
'Seeds 2': 'seeds_2',
|
|
'seeds_2': 'seeds_2',
|
|
'Seeds 3': 'seeds_3',
|
|
'seeds_3': 'seeds_3',
|
|
};
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// Known scheduler suffixes (ComfyUI KSampler.SCHEDULERS)
|
|
// Sorted by length (descending) for longest-match-first parsing.
|
|
// ---------------------------------------------------------------------------
|
|
const SCHEDULER_SUFFIXES = [
|
|
'sgm_uniform',
|
|
'ddim_uniform',
|
|
'linear_quadratic',
|
|
'kl_optimal',
|
|
'exponential',
|
|
'karras',
|
|
'simple',
|
|
'normal',
|
|
'beta',
|
|
];
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// Scheduler-only values (values that are schedulers, not samplers)
|
|
// ---------------------------------------------------------------------------
|
|
const SCHEDULER_ONLY_VALUES = new Set([
|
|
'simple', 'sgm_uniform', 'karras', 'exponential',
|
|
'ddim_uniform', 'beta', 'normal', 'linear_quadratic', 'kl_optimal',
|
|
]);
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// Param key → widget name candidates (searched in order)
|
|
// ---------------------------------------------------------------------------
|
|
const PARAM_TO_WIDGET_CANDIDATES = {
|
|
seed: ['seed', 'noise_seed'],
|
|
steps: ['steps'],
|
|
cfg: ['cfg'],
|
|
sampler: ['sampler_name', 'sampler'],
|
|
scheduler: ['scheduler'],
|
|
};
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// Parse a combined sampler+scheduler value (space-separated or underscore)
|
|
// e.g., "Euler a Karras", "DPM++ 2M beta", "er_sde_beta"
|
|
// Returns { sampler: internalName|null, scheduler: internalName|null } or null
|
|
// ---------------------------------------------------------------------------
|
|
function parseCombinedSamplerName(rawValue) {
|
|
if (!rawValue || typeof rawValue !== 'string') return null;
|
|
const trimmed = rawValue.trim();
|
|
if (!trimmed) return null;
|
|
|
|
// Try space-separated first: split on last space
|
|
const spaceIdx = trimmed.lastIndexOf(' ');
|
|
if (spaceIdx > 0) {
|
|
const candidateScheduler = trimmed.slice(spaceIdx + 1).trim().toLowerCase();
|
|
if (SCHEDULER_SUFFIXES.includes(candidateScheduler)) {
|
|
const samplerPart = trimmed.slice(0, spaceIdx).trim();
|
|
const internalSampler = SAMPLER_DISPLAY_TO_INTERNAL[samplerPart];
|
|
if (internalSampler) {
|
|
return { sampler: internalSampler, scheduler: candidateScheduler };
|
|
}
|
|
// samplerPart might be a combined name itself (e.g., "DPM++ 2M SDE")
|
|
// Try recursing (one level max) — already handled since we split at last space
|
|
}
|
|
}
|
|
|
|
// Try underscore-separated: e.g., "er_sde_beta"
|
|
const underIdx = trimmed.lastIndexOf('_');
|
|
if (underIdx > 0) {
|
|
const candidateScheduler = trimmed.slice(underIdx + 1).trim().toLowerCase();
|
|
if (SCHEDULER_SUFFIXES.includes(candidateScheduler)) {
|
|
const samplerPart = trimmed.slice(0, underIdx).trim();
|
|
const internalSampler = SAMPLER_DISPLAY_TO_INTERNAL[samplerPart] || SAMPLER_DISPLAY_TO_INTERNAL[samplerPart.toLowerCase()];
|
|
if (internalSampler) {
|
|
return { sampler: internalSampler, scheduler: candidateScheduler };
|
|
}
|
|
}
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// Main resolver: takes a raw sampler value from recipe/showcase metadata
|
|
// and returns { sampler: internalName|null, scheduler: internalName|null }
|
|
// ---------------------------------------------------------------------------
|
|
function resolveSamplerScheduler(rawValue) {
|
|
if (!rawValue || typeof rawValue !== 'string') {
|
|
return { sampler: null, scheduler: null };
|
|
}
|
|
|
|
const trimmed = rawValue.trim();
|
|
if (!trimmed) return { sampler: null, scheduler: null };
|
|
|
|
// 1. Try direct lookup first
|
|
const direct = SAMPLER_DISPLAY_TO_INTERNAL[trimmed];
|
|
if (direct) return { sampler: direct, scheduler: null };
|
|
|
|
// 2. Try lowercase direct lookup
|
|
const lowerDirect = SAMPLER_DISPLAY_TO_INTERNAL[trimmed.toLowerCase()];
|
|
if (lowerDirect) return { sampler: lowerDirect, scheduler: null };
|
|
|
|
// 3. Scheduler-only value? (check BEFORE the "already internal name" regex,
|
|
// because scheduler values like "karras", "simple" also match that pattern)
|
|
if (SCHEDULER_ONLY_VALUES.has(trimmed.toLowerCase())) {
|
|
return { sampler: null, scheduler: trimmed.toLowerCase() };
|
|
}
|
|
|
|
// 4. Already an internal name? (lowercase, no spaces)
|
|
if (/^[a-z][a-z0-9_]+$/.test(trimmed)) {
|
|
return { sampler: trimmed, scheduler: null };
|
|
}
|
|
|
|
// 5. Try combined name parsing (space-separated or underscore)
|
|
const combined = parseCombinedSamplerName(trimmed);
|
|
if (combined) return combined;
|
|
|
|
// 6. Custom format like "multistep/dpmpp_2m_simple" — try extracting the last segment
|
|
if (trimmed.includes('/')) {
|
|
const parts = trimmed.split('/');
|
|
const last = parts[parts.length - 1];
|
|
if (last) {
|
|
const subResult = resolveSamplerScheduler(last);
|
|
if (subResult.sampler || subResult.scheduler) return subResult;
|
|
}
|
|
}
|
|
|
|
// 7. Unrecognized — return null for both
|
|
return { sampler: null, scheduler: null };
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// Find which gen params can be sent to a given node, matching by widget names
|
|
// Returns array of { widgetName, value } objects
|
|
// ---------------------------------------------------------------------------
|
|
function findMatchingWidgets(nodeWidgetNames, resolvedParams) {
|
|
if (!nodeWidgetNames || !Array.isArray(nodeWidgetNames) || nodeWidgetNames.length === 0) {
|
|
return [];
|
|
}
|
|
|
|
const widgetSet = new Set(nodeWidgetNames.map(w => String(w).toLowerCase()));
|
|
const updates = [];
|
|
|
|
// Simple numeric/string params: seed, steps, cfg
|
|
const simpleParams = [
|
|
{ key: 'seed', value: resolvedParams.seed },
|
|
{ key: 'steps', value: resolvedParams.steps },
|
|
{ key: 'cfg', value: resolvedParams.cfg },
|
|
];
|
|
for (const { key, value } of simpleParams) {
|
|
if (value === undefined || value === null || value === '') continue;
|
|
const candidates = PARAM_TO_WIDGET_CANDIDATES[key] || [key];
|
|
for (const candidate of candidates) {
|
|
if (widgetSet.has(candidate.toLowerCase())) {
|
|
updates.push({ widgetName: candidate, value: String(value) });
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Sampler
|
|
if (resolvedParams.sampler) {
|
|
const candidates = PARAM_TO_WIDGET_CANDIDATES.sampler;
|
|
for (const candidate of candidates) {
|
|
if (widgetSet.has(candidate.toLowerCase())) {
|
|
updates.push({ widgetName: candidate, value: resolvedParams.sampler });
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Scheduler
|
|
if (resolvedParams.scheduler) {
|
|
const candidates = PARAM_TO_WIDGET_CANDIDATES.scheduler;
|
|
for (const candidate of candidates) {
|
|
if (widgetSet.has(candidate.toLowerCase())) {
|
|
updates.push({ widgetName: candidate, value: resolvedParams.scheduler });
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
return updates;
|
|
}
|
|
|
|
export {
|
|
SAMPLER_DISPLAY_TO_INTERNAL,
|
|
SCHEDULER_SUFFIXES,
|
|
SCHEDULER_ONLY_VALUES,
|
|
PARAM_TO_WIDGET_CANDIDATES,
|
|
parseCombinedSamplerName,
|
|
resolveSamplerScheduler,
|
|
findMatchingWidgets,
|
|
};
|