Skip to content

GitLab

"sgl-router/src/git@developer.sourcefind.cn:change/sglang.git" did not exist on "38907fe639047fa21dfa22eadbeb7512b1ecd053"
Commit 85865158 authored by comfyanonymous's avatar comfyanonymous
Browse files

Move the CheckpointLoaderSimple to the right section.

parent 42152062
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
nodes.py
View file @ 85865158
...@@ -213,7 +213,7 @@ class CheckpointLoaderSimple: ...@@ -213,7 +213,7 @@ class CheckpointLoaderSimple:
RETURN_TYPES = ("MODEL", "CLIP", "VAE") RETURN_TYPES = ("MODEL", "CLIP", "VAE")
FUNCTION = "load_checkpoint" FUNCTION = "load_checkpoint"
CATEGORY = "_for_testing" CATEGORY = "loaders"
def load_checkpoint(self, ckpt_name, output_vae=True, output_clip=True): def load_checkpoint(self, ckpt_name, output_vae=True, output_clip=True):
ckpt_path = os.path.join(self.ckpt_dir, ckpt_name) ckpt_path = os.path.join(self.ckpt_dir, ckpt_name)
...@@ -844,6 +844,7 @@ class ImageInvert: ...@@ -844,6 +844,7 @@ class ImageInvert:
NODE_CLASS_MAPPINGS = { NODE_CLASS_MAPPINGS = {
"KSampler": KSampler, "KSampler": KSampler,
"CheckpointLoader": CheckpointLoader, "CheckpointLoader": CheckpointLoader,
"CheckpointLoaderSimple": CheckpointLoaderSimple,
"CLIPTextEncode": CLIPTextEncode, "CLIPTextEncode": CLIPTextEncode,
"CLIPSetLastLayer": CLIPSetLastLayer, "CLIPSetLastLayer": CLIPSetLastLayer,
"VAEDecode": VAEDecode, "VAEDecode": VAEDecode,
...@@ -872,7 +873,6 @@ NODE_CLASS_MAPPINGS = { ...@@ -872,7 +873,6 @@ NODE_CLASS_MAPPINGS = {
"DiffControlNetLoader": DiffControlNetLoader, "DiffControlNetLoader": DiffControlNetLoader,
"T2IAdapterLoader": T2IAdapterLoader, "T2IAdapterLoader": T2IAdapterLoader,
"VAEDecodeTiled": VAEDecodeTiled, "VAEDecodeTiled": VAEDecodeTiled,
"CheckpointLoaderSimple": CheckpointLoaderSimple,
} }
CUSTOM_NODE_PATH = os.path.join(os.path.dirname(os.path.realpath(__file__)), "custom_nodes") CUSTOM_NODE_PATH = os.path.join(os.path.dirname(os.path.realpath(__file__)), "custom_nodes")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment