mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-08-06 22:10:14 -03:00
fix(metadata-overwrite): rename 'checkpoint' input to 'model'
This commit is contained in:
@@ -14,7 +14,7 @@ OVERWRITE = "overwrite" # Manual metadata overwrite from MetadataOverwriteLM no
|
|||||||
# Field names that the MetadataOverwriteLM node and its extractor share
|
# Field names that the MetadataOverwriteLM node and its extractor share
|
||||||
METADATA_OVERWRITE_FIELDS = (
|
METADATA_OVERWRITE_FIELDS = (
|
||||||
"prompt", "negative_prompt", "seed", "steps", "cfg_scale",
|
"prompt", "negative_prompt", "seed", "steps", "cfg_scale",
|
||||||
"sampler", "scheduler", "checkpoint", "loras", "size",
|
"sampler", "scheduler", "model", "loras", "size",
|
||||||
"clip_skip", "additional_data",
|
"clip_skip", "additional_data",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
@@ -681,6 +681,13 @@ class MetadataProcessor:
|
|||||||
if value: # truthy check — only overwrite when user provided a real value
|
if value: # truthy check — only overwrite when user provided a real value
|
||||||
params[key] = value
|
params[key] = value
|
||||||
|
|
||||||
|
# Bridge: the overwrite node exposes the field as "model" (more accurate),
|
||||||
|
# but the internal pipeline key remains "checkpoint" for backward compatibility
|
||||||
|
# with A1111 metadata format and downstream consumers.
|
||||||
|
if params.get("model"):
|
||||||
|
params["checkpoint"] = params["model"]
|
||||||
|
del params["model"]
|
||||||
|
|
||||||
return params
|
return params
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
|
|||||||
@@ -81,11 +81,14 @@ class MetadataOverwriteLM:
|
|||||||
"tooltip": "Scheduler name. Only overwrites when non-empty.",
|
"tooltip": "Scheduler name. Only overwrites when non-empty.",
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
"checkpoint": (
|
"model": (
|
||||||
"STRING",
|
"STRING",
|
||||||
{
|
{
|
||||||
"default": "",
|
"default": "",
|
||||||
"tooltip": "Checkpoint / model name. Only overwrites when non-empty.",
|
"tooltip": (
|
||||||
|
"The checkpoint or diffusion model (UNet) used "
|
||||||
|
"for generation. Only overwrites when non-empty."
|
||||||
|
),
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
"loras": (
|
"loras": (
|
||||||
|
|||||||
@@ -843,7 +843,7 @@ def test_metadata_overwrite_extractor_stores_truthy_values(metadata_registry):
|
|||||||
"cfg_scale": 7.5,
|
"cfg_scale": 7.5,
|
||||||
"sampler": "",
|
"sampler": "",
|
||||||
"scheduler": "",
|
"scheduler": "",
|
||||||
"checkpoint": "myModel.safetensors",
|
"model": "myModel.safetensors",
|
||||||
"loras": "<lora:detail:0.8>",
|
"loras": "<lora:detail:0.8>",
|
||||||
"size": "1024x768",
|
"size": "1024x768",
|
||||||
"clip_skip": 0,
|
"clip_skip": 0,
|
||||||
@@ -860,7 +860,7 @@ def test_metadata_overwrite_extractor_stores_truthy_values(metadata_registry):
|
|||||||
assert params["prompt"] == "a beautiful landscape"
|
assert params["prompt"] == "a beautiful landscape"
|
||||||
assert params["seed"] == 42
|
assert params["seed"] == 42
|
||||||
assert params["cfg_scale"] == 7.5
|
assert params["cfg_scale"] == 7.5
|
||||||
assert params["checkpoint"] == "myModel.safetensors"
|
assert params["model"] == "myModel.safetensors"
|
||||||
assert params["loras"] == "<lora:detail:0.8>"
|
assert params["loras"] == "<lora:detail:0.8>"
|
||||||
assert params["size"] == "1024x768"
|
assert params["size"] == "1024x768"
|
||||||
assert params["additional_data"] == '{"Copyright": "CC0"}'
|
assert params["additional_data"] == '{"Copyright": "CC0"}'
|
||||||
@@ -986,7 +986,7 @@ def test_fill_missing_metadata_skips_overwrite_for_bypassed_node(metadata_regist
|
|||||||
"ow-1",
|
"ow-1",
|
||||||
"MetadataOverwriteLM",
|
"MetadataOverwriteLM",
|
||||||
{"seed": 99, "prompt": "test", "steps": 0, "cfg_scale": 0.0,
|
{"seed": 99, "prompt": "test", "steps": 0, "cfg_scale": 0.0,
|
||||||
"negative_prompt": "", "sampler": "", "scheduler": "", "checkpoint": "",
|
"negative_prompt": "", "sampler": "", "scheduler": "", "model": "",
|
||||||
"loras": "", "size": "", "clip_skip": 0, "additional_data": ""},
|
"loras": "", "size": "", "clip_skip": 0, "additional_data": ""},
|
||||||
None,
|
None,
|
||||||
)
|
)
|
||||||
@@ -1018,7 +1018,7 @@ def test_fill_missing_metadata_fills_overwrite_for_muted_node(metadata_registry)
|
|||||||
"ow-1",
|
"ow-1",
|
||||||
"MetadataOverwriteLM",
|
"MetadataOverwriteLM",
|
||||||
{"seed": 88, "prompt": "test2", "steps": 0, "cfg_scale": 0.0,
|
{"seed": 88, "prompt": "test2", "steps": 0, "cfg_scale": 0.0,
|
||||||
"negative_prompt": "", "sampler": "", "scheduler": "", "checkpoint": "",
|
"negative_prompt": "", "sampler": "", "scheduler": "", "model": "",
|
||||||
"loras": "", "size": "", "clip_skip": 0, "additional_data": ""},
|
"loras": "", "size": "", "clip_skip": 0, "additional_data": ""},
|
||||||
None,
|
None,
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user