mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-08-07 14:30:15 -03:00
191c4e03cd
The model field now accepts either a manual string or a MODEL connection. When wired, the model name is extracted from the patcher's cached_patcher_init (registered by core loaders load_checkpoint_guess_config and load_diffusion_model, preserved through LoRA clones) and converted to a ComfyUI-style relative name via config model roots. - model input declared as "STRING,MODEL" with widgetType STRING, so the text widget and the dual-type connection slot coexist; non-STRING/MODEL links are rejected by frontend and backend type validation - UNETLoaderLM GGUF branch now registers a custom cached_patcher_init reload factory so GGUF models participate in name extraction and ModelPatcher deepclone/dynamic machinery - shared collect_overwrite_params() helper keeps the node and the metadata extractor conversion logic in sync; extraction failures are logged instead of silently dropping the overwrite
43 lines
1.6 KiB
Python
43 lines
1.6 KiB
Python
"""Shared helpers for Metadata Overwrite node metadata collection.
|
|
|
|
Used by both the MetadataOverwriteLM node (execution time) and the
|
|
MetadataOverwriteExtractor (hook time) so the conversion/filtering logic
|
|
cannot drift between the two paths.
|
|
"""
|
|
|
|
import logging
|
|
from typing import Any, Dict
|
|
|
|
from ..utils.utils import model_patcher_to_name
|
|
from .constants import CLIP_SKIP_SENTINEL, METADATA_OVERWRITE_FIELDS
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def collect_overwrite_params(values: Dict[str, Any]) -> Dict[str, Any]:
|
|
"""Convert node input values into non-default overwrite parameters.
|
|
|
|
For most fields, a falsy value (empty string, 0) means "not set" and is
|
|
skipped. clip_skip uses a dedicated sentinel (-25) so that a wired value
|
|
of 0 is preserved. The ``model`` field accepts either a manual string or
|
|
a wired MODEL (ModelPatcher) connection; in the latter case the source
|
|
model name is extracted from the patcher's ``cached_patcher_init`` and
|
|
stored as a ComfyUI-style relative path.
|
|
"""
|
|
result: Dict[str, Any] = {}
|
|
for key in METADATA_OVERWRITE_FIELDS:
|
|
value = values.get(key)
|
|
if key == "model" and not isinstance(value, str):
|
|
value = model_patcher_to_name(value)
|
|
if value is None:
|
|
logger.warning(
|
|
"Could not extract model name from wired MODEL input "
|
|
"(no cached_patcher_init); model metadata overwrite skipped"
|
|
)
|
|
if key == "clip_skip":
|
|
if value != CLIP_SKIP_SENTINEL:
|
|
result[key] = value
|
|
elif value:
|
|
result[key] = value
|
|
return result
|