diff --git a/nodes.py b/nodes.py index 30eafb3..23b8cf5 100644 --- a/nodes.py +++ b/nodes.py @@ -213,7 +213,7 @@ class CheckpointLoaderSimple: RETURN_TYPES = ("MODEL", "CLIP", "VAE") FUNCTION = "load_checkpoint" - CATEGORY = "_for_testing" + CATEGORY = "loaders" def load_checkpoint(self, ckpt_name, output_vae=True, output_clip=True): ckpt_path = os.path.join(self.ckpt_dir, ckpt_name) @@ -844,6 +844,7 @@ class ImageInvert: NODE_CLASS_MAPPINGS = { "KSampler": KSampler, "CheckpointLoader": CheckpointLoader, + "CheckpointLoaderSimple": CheckpointLoaderSimple, "CLIPTextEncode": CLIPTextEncode, "CLIPSetLastLayer": CLIPSetLastLayer, "VAEDecode": VAEDecode, @@ -872,7 +873,6 @@ NODE_CLASS_MAPPINGS = { "DiffControlNetLoader": DiffControlNetLoader, "T2IAdapterLoader": T2IAdapterLoader, "VAEDecodeTiled": VAEDecodeTiled, - "CheckpointLoaderSimple": CheckpointLoaderSimple, } CUSTOM_NODE_PATH = os.path.join(os.path.dirname(os.path.realpath(__file__)), "custom_nodes")