diff --git a/latent/nodes.py b/latent/nodes.py index 43a7c0f..5f3b09b 100644 --- a/latent/nodes.py +++ b/latent/nodes.py @@ -93,8 +93,8 @@ class LLSSimpleEmptyLatent: "width": ("INT", {"default": 512, "min": 64, "max": 8192, "step": 8}), "height": ("INT", {"default": 512, "min": 64, "max": 8192, "step": 8}), "batch_size": ("INT", {"default": 1, "min": 1, "max": 64}), - "resize_mode": (_RESIZE_MODES, {"default": "keep_aspect"}), "model_family": (MODEL_FAMILY_CHOICES, {"default": "Auto"}), + "resize_mode": (_RESIZE_MODES, {"default": "keep_aspect"}), }, "optional": { "model": ("MODEL",), diff --git a/tests/test_loader_prompt_refactor.py b/tests/test_loader_prompt_refactor.py index 1416a02..a143638 100644 --- a/tests/test_loader_prompt_refactor.py +++ b/tests/test_loader_prompt_refactor.py @@ -449,6 +449,17 @@ class TestLoaderPromptRefactor(unittest.TestCase): required = node_cls.INPUT_TYPES()["required"] optional = node_cls.INPUT_TYPES()["optional"] + self.assertEqual( + tuple(required.keys()), + ( + "size_preset", + "width", + "height", + "batch_size", + "model_family", + "resize_mode", + ), + ) self.assertEqual(required["resize_mode"][0], ["keep_aspect", "crop_center", "stretch", "none"]) self.assertEqual(optional["image"][0], "IMAGE") self.assertEqual(optional["vae"][0], "VAE")