修改laten出现BUG
This commit is contained in:
+1
-1
@@ -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",),
|
||||
|
||||
@@ -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")
|
||||
|
||||
Reference in New Issue
Block a user