From 85865158869a67b15012b64b79a080f8c2f9f6b0 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Fri, 3 Mar 2023 13:09:44 -0500
Subject: [PATCH] Move the CheckpointLoaderSimple to the right section.

---
 nodes.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/nodes.py b/nodes.py
index 30eafb35..23b8cf55 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")