DemonGatanjieuandClaude Opus 5.5
c59927e701
AMB read models only from ordinary drop-downs, so rgthree's Power Lora Loader (rows are
{on, lora, strength} objects) was invisible: Model Check never reported its missing LoRAs.
Now its switched-on rows are Model Check boxes (lora_nodes.js powerLoraRowBoxes: ComfyUI's LoRA
list as choices, a fix changes only the row's LoRA; downloads put files in the same way), saved
fingerprints and a picture's model search read them, the workflow summary lists them, and saved
model references include them and Easy-Use LoRA stack rows (nodeModelValues /
_model_reference_values, mirrored). Only these widely used nodes; ComfyUI's loaders stay the
reference.
recipe_parser.js was over 600 lines: which LoRAs a node loads moves to recipe_loras.js and the
box readers to recipe_widgets.js.
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
62 lines
3.0 KiB
Python
62 lines
3.0 KiB
Python
from pathlib import Path
|
|
import sys
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
PLUGIN = Path(__file__).resolve().parents[1]
|
|
sys.path[:0] = [str(PLUGIN), str(PLUGIN.parents[1])]
|
|
from api import recipe_schema
|
|
|
|
|
|
class RecipeModelReferenceTests(unittest.TestCase):
|
|
def _references(self, nodes):
|
|
with mock.patch.object(recipe_schema, '_identity_for_reference', return_value=({"status": "unavailable"}, None)):
|
|
return recipe_schema._build_model_references({"workflow": {"nodes": nodes}, "params": {}})
|
|
|
|
def test_all_in_one_loaders_use_verified_layouts_and_skip_placeholders(self):
|
|
refs = self._references([
|
|
{"id": 1, "type": "easy a1111Loader", "widgets_values": ["base.safetensors", "Baked VAE", -2, "style.safetensors"]},
|
|
{"id": 2, "type": "easy fullLoader", "widgets_values": ["full.safetensors", "Default", "vae.safetensors", -2, "None"]},
|
|
])
|
|
self.assertEqual(
|
|
[(r["node_id"], r["widget_index"], r["category"], r["widget_name"], r["saved_value"]) for r in refs],
|
|
[
|
|
(1, 0, "checkpoint", "ckpt_name", "base.safetensors"),
|
|
(1, 3, "lora", "lora_name", "style.safetensors"),
|
|
(2, 0, "checkpoint", "ckpt_name", "full.safetensors"),
|
|
(2, 2, "vae", "vae_name", "vae.safetensors"),
|
|
],
|
|
)
|
|
|
|
def test_native_labels_and_unknown_nodes_are_unchanged(self):
|
|
refs = self._references([
|
|
{"id": 3, "type": "CheckpointLoaderSimple", "widgets_values": ["sdxl.safetensors"]},
|
|
{"id": 4, "type": "SomeCustomLoader", "widgets_values": ["other_vae.safetensors"]},
|
|
])
|
|
self.assertEqual([(r["node_id"], r["category"], r["widget_name"]) for r in refs], [(3, "checkpoint", "checkpoint")])
|
|
|
|
def test_stack_nodes_by_their_rows(self):
|
|
# rgthree's Power Lora Loader: rows are dicts, numbered as rgthree numbers them; off rows load nothing.
|
|
# Easy-Use's LoRA stack: lora_N_name at 3 + 4*(N-1), up to num_loras; "None" is no model.
|
|
refs = self._references([
|
|
{"id": 5, "type": "Power Lora Loader (rgthree)", "widgets_values": [
|
|
{}, {"type": "PowerLoraLoaderHeaderWidget"},
|
|
{"on": True, "lora": "a.safetensors", "strength": 1},
|
|
{"on": False, "lora": "off.safetensors", "strength": 1},
|
|
{"on": True, "lora": "b.safetensors", "strength": 0.5}, {}, ""]},
|
|
{"id": 6, "type": "easy loraStack", "widgets_values": [
|
|
True, "simple", 2, "c.safetensors", 1, 1, 1, "None", 1, 1, 1, "unused.safetensors"]},
|
|
])
|
|
self.assertEqual(
|
|
[(r["node_id"], r["widget_index"], r["category"], r["widget_name"], r["saved_value"]) for r in refs],
|
|
[
|
|
(5, 2, "lora", "lora_1", "a.safetensors"),
|
|
(5, 4, "lora", "lora_3", "b.safetensors"),
|
|
(6, 3, "lora", "lora_1_name", "c.safetensors"),
|
|
],
|
|
)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|