feat: add model loading main category mapping

Yourz · Yourz · commit 6954e3646c61 · 2026-02-16T15:08:20.000+08:00
diff --git a/comfy_extras/nodes_audio_encoder.py b/comfy_extras/nodes_audio_encoder.py
@@ -11,6 +11,7 @@ def define_schema(cls) -> io.Schema:
         return io.Schema(
             node_id="AudioEncoderLoader",
             category="loaders",
+            main_category="Model Loading",
             inputs=[
                 io.Combo.Input(
                     "audio_encoder_name",
diff --git a/comfy_extras/nodes_chroma_radiance.py b/comfy_extras/nodes_chroma_radiance.py
@@ -34,6 +34,7 @@ def define_schema(cls) -> io.Schema:
         return io.Schema(
             node_id="ChromaRadianceOptions",
             category="model_patches/chroma_radiance",
+            main_category="Model Loading",
             description="Allows setting advanced options for the Chroma Radiance model.",
             inputs=[
                 io.Model.Input(id="model"),
diff --git a/comfy_extras/nodes_eps.py b/comfy_extras/nodes_eps.py
@@ -19,6 +19,7 @@ def define_schema(cls):
         return io.Schema(
             node_id="Epsilon Scaling",
             category="model_patches/unet",
+            main_category="Model Loading",
             inputs=[
                 io.Model.Input("model"),
                 io.Float.Input(
@@ -84,6 +85,7 @@ def define_schema(cls):
             node_id="TemporalScoreRescaling",
             display_name="TSR - Temporal Score Rescaling",
             category="model_patches/unet",
+            main_category="Model Loading",
             inputs=[
                 io.Model.Input("model"),
                 io.Float.Input(
diff --git a/comfy_extras/nodes_flux.py b/comfy_extras/nodes_flux.py
@@ -61,6 +61,7 @@ def define_schema(cls):
         return io.Schema(
             node_id="FluxGuidance",
             category="advanced/conditioning/flux",
+            main_category="Model Loading",
             inputs=[
                 io.Conditioning.Input("conditioning"),
                 io.Float.Input("guidance", default=3.5, min=0.0, max=100.0, step=0.1),
diff --git a/comfy_extras/nodes_freelunch.py b/comfy_extras/nodes_freelunch.py
@@ -30,6 +30,7 @@ def define_schema(cls):
         return IO.Schema(
             node_id="FreeU",
             category="model_patches/unet",
+            main_category="Model Loading",
             inputs=[
                 IO.Model.Input("model"),
                 IO.Float.Input("b1", default=1.1, min=0.0, max=10.0, step=0.01),
@@ -77,6 +78,7 @@ def define_schema(cls):
         return IO.Schema(
             node_id="FreeU_V2",
             category="model_patches/unet",
+            main_category="Model Loading",
             inputs=[
                 IO.Model.Input("model"),
                 IO.Float.Input("b1", default=1.3, min=0.0, max=10.0, step=0.01),
diff --git a/comfy_extras/nodes_hidream.py b/comfy_extras/nodes_hidream.py
@@ -12,6 +12,7 @@ def define_schema(cls):
         return io.Schema(
             node_id="QuadrupleCLIPLoader",
             category="advanced/loaders",
+            main_category="Model Loading",
             description="[Recipes]\n\nhidream: long clip-l, long clip-g, t5xxl, llama_8b_3.1_instruct",
             inputs=[
                 io.Combo.Input("clip_name1", options=folder_paths.get_filename_list("text_encoders")),
diff --git a/comfy_extras/nodes_hunyuan.py b/comfy_extras/nodes_hunyuan.py
@@ -177,6 +177,7 @@ def define_schema(cls):
             node_id="LatentUpscaleModelLoader",
             display_name="Load Latent Upscale Model",
             category="loaders",
+            main_category="Model Loading",
             inputs=[
                 io.Combo.Input("model_name", options=folder_paths.get_filename_list("latent_upscale_models")),
             ],
diff --git a/comfy_extras/nodes_hypernetwork.py b/comfy_extras/nodes_hypernetwork.py
@@ -103,6 +103,7 @@ def define_schema(cls):
         return IO.Schema(
             node_id="HypernetworkLoader",
             category="loaders",
+            main_category="Model Loading",
             inputs=[
                 IO.Model.Input("model"),
                 IO.Combo.Input("hypernetwork_name", options=folder_paths.get_filename_list("hypernetworks")),
diff --git a/comfy_extras/nodes_hypertile.py b/comfy_extras/nodes_hypertile.py
@@ -28,6 +28,7 @@ def define_schema(cls):
         return io.Schema(
             node_id="HyperTile",
             category="model_patches/unet",
+            main_category="Model Loading",
             inputs=[
                 io.Model.Input("model"),
                 io.Int.Input("tile_size", default=256, min=1, max=2048),
diff --git a/comfy_extras/nodes_lt_audio.py b/comfy_extras/nodes_lt_audio.py
@@ -176,6 +176,7 @@ def define_schema(cls) -> io.Schema:
             node_id="LTXAVTextEncoderLoader",
             display_name="LTXV Audio Text Encoder Loader",
             category="advanced/loaders",
+            main_category="Model Loading",
             description="[Recipes]\n\nltxav: gemma 3 12B",
             inputs=[
                 io.Combo.Input(
diff --git a/comfy_extras/nodes_model_downscale.py b/comfy_extras/nodes_model_downscale.py
@@ -11,6 +11,7 @@ def define_schema(cls):
             node_id="PatchModelAddDownscale",
             display_name="PatchModelAddDownscale (Kohya Deep Shrink)",
             category="model_patches/unet",
+            main_category="Model Loading",
             inputs=[
                 io.Model.Input("model"),
                 io.Int.Input("block_number", default=3, min=1, max=32, step=1),
diff --git a/comfy_extras/nodes_model_patch.py b/comfy_extras/nodes_model_patch.py
@@ -230,6 +230,7 @@ def INPUT_TYPES(s):
     EXPERIMENTAL = True
 
     CATEGORY = "advanced/loaders"
+    MAIN_CATEGORY = "Model Loading"
 
     def load_model_patch(self, name):
         model_patch_path = folder_paths.get_full_path_or_raise("model_patches", name)
@@ -456,6 +457,7 @@ def INPUT_TYPES(s):
     EXPERIMENTAL = True
 
     CATEGORY = "advanced/loaders/qwen"
+    MAIN_CATEGORY = "Model Loading"
 
     def diffsynth_controlnet(self, model, model_patch, vae, image=None, strength=1.0, inpaint_image=None, mask=None):
         model_patched = model.clone()
@@ -489,6 +491,7 @@ def INPUT_TYPES(s):
                 "optional": {"image": ("IMAGE",), "inpaint_image": ("IMAGE",), "mask": ("MASK",)}}
 
     CATEGORY = "advanced/loaders/zimage"
+    MAIN_CATEGORY = "Model Loading"
 
 class UsoStyleProjectorPatch:
     def __init__(self, model_patch, encoded_image):
@@ -525,6 +528,7 @@ def INPUT_TYPES(s):
     EXPERIMENTAL = True
 
     CATEGORY = "advanced/model_patches/flux"
+    MAIN_CATEGORY = "Model Loading"
 
     def apply_patch(self, model, model_patch, clip_vision_output):
         encoded_image = torch.stack((clip_vision_output.all_hidden_states[:, -20], clip_vision_output.all_hidden_states[:, -11], clip_vision_output.penultimate_hidden_states))
diff --git a/comfy_extras/nodes_pag.py b/comfy_extras/nodes_pag.py
@@ -16,6 +16,7 @@ def define_schema(cls):
         return io.Schema(
             node_id="PerturbedAttentionGuidance",
             category="model_patches/unet",
+            main_category="Model Loading",
             inputs=[
                 io.Model.Input("model"),
                 io.Float.Input("scale", default=3.0, min=0.0, max=100.0, step=0.01, round=0.01),
diff --git a/comfy_extras/nodes_rope.py b/comfy_extras/nodes_rope.py
@@ -8,6 +8,7 @@ def define_schema(cls):
         return io.Schema(
             node_id="ScaleROPE",
             category="advanced/model_patches",
+            main_category="Model Loading",
             description="Scale and shift the ROPE of the model.",
             is_experimental=True,
             inputs=[
diff --git a/comfy_extras/nodes_sd3.py b/comfy_extras/nodes_sd3.py
@@ -14,6 +14,7 @@ def define_schema(cls):
         return io.Schema(
             node_id="TripleCLIPLoader",
             category="advanced/loaders",
+            main_category="Model Loading",
             description="[Recipes]\n\nsd3: clip-l, clip-g, t5",
             inputs=[
                 io.Combo.Input("clip_name1", options=folder_paths.get_filename_list("text_encoders")),
diff --git a/comfy_extras/nodes_tomesd.py b/comfy_extras/nodes_tomesd.py
@@ -152,6 +152,7 @@ def define_schema(cls):
         return io.Schema(
             node_id="TomePatchModel",
             category="model_patches/unet",
+            main_category="Model Loading",
             inputs=[
                 io.Model.Input("model"),
                 io.Float.Input("ratio", default=0.3, min=0.0, max=1.0, step=0.01),
diff --git a/comfy_extras/nodes_train.py b/comfy_extras/nodes_train.py
@@ -1151,6 +1151,7 @@ def define_schema(cls):
             node_id="LoraModelLoader",
             display_name="Load LoRA Model",
             category="loaders",
+            main_category="Model Loading",
             is_experimental=True,
             inputs=[
                 io.Model.Input(
diff --git a/comfy_extras/nodes_upscale_model.py b/comfy_extras/nodes_upscale_model.py
@@ -22,6 +22,7 @@ def define_schema(cls):
             node_id="UpscaleModelLoader",
             display_name="Load Upscale Model",
             category="loaders",
+            main_category="Model Loading",
             inputs=[
                 io.Combo.Input("model_name", options=folder_paths.get_filename_list("upscale_models")),
             ],
diff --git a/comfy_extras/nodes_video_model.py b/comfy_extras/nodes_video_model.py
@@ -16,6 +16,7 @@ def INPUT_TYPES(s):
     FUNCTION = "load_checkpoint"
 
     CATEGORY = "loaders/video_models"
+    MAIN_CATEGORY = "Model Loading"
 
     def load_checkpoint(self, ckpt_name, output_vae=True, output_clip=True):
         ckpt_path = folder_paths.get_full_path_or_raise("checkpoints", ckpt_name)
diff --git a/nodes.py b/nodes.py
@@ -568,6 +568,7 @@ def VALIDATE_INPUTS(s, latent):
 
 class CheckpointLoader:
     SEARCH_ALIASES = ["load model", "model loader"]
+    MAIN_CATEGORY = "Model Loading"
 
     @classmethod
     def INPUT_TYPES(s):
@@ -585,6 +586,8 @@ def load_checkpoint(self, config_name, ckpt_name):
         return comfy.sd.load_checkpoint(config_path, ckpt_path, output_vae=True, output_clip=True, embedding_directory=folder_paths.get_folder_paths("embeddings"))
 
 class CheckpointLoaderSimple:
+    MAIN_CATEGORY = "Model Loading"
+
     @classmethod
     def INPUT_TYPES(s):
         return {
@@ -609,6 +612,7 @@ def load_checkpoint(self, ckpt_name):
 
 class DiffusersLoader:
     SEARCH_ALIASES = ["load diffusers model"]
+    MAIN_CATEGORY = "Model Loading"
 
     @classmethod
     def INPUT_TYPES(cls):
@@ -637,6 +641,8 @@ def load_checkpoint(self, model_path, output_vae=True, output_clip=True):
 
 
 class unCLIPCheckpointLoader:
+    MAIN_CATEGORY = "Model Loading"
+
     @classmethod
     def INPUT_TYPES(s):
         return {"required": { "ckpt_name": (folder_paths.get_filename_list("checkpoints"), ),
@@ -669,6 +675,7 @@ def set_last_layer(self, clip, stop_at_clip_layer):
 
 class LoraLoader:
     ESSENTIALS_CATEGORY = "Image Generation"
+    MAIN_CATEGORY = "Model Loading"
 
     def __init__(self):
         self.loaded_lora = None
@@ -713,6 +720,8 @@ def load_lora(self, model, clip, lora_name, strength_model, strength_clip):
         return (model_lora, clip_lora)
 
 class LoraLoaderModelOnly(LoraLoader):
+    MAIN_CATEGORY = "Model Loading"
+
     @classmethod
     def INPUT_TYPES(s):
         return {"required": { "model": ("MODEL",),
@@ -728,6 +737,8 @@ def load_lora_model_only(self, model, lora_name, strength_model):
 class VAELoader:
     video_taes = ["taehv", "lighttaew2_2", "lighttaew2_1", "lighttaehy1_5", "taeltx_2"]
     image_taes = ["taesd", "taesdxl", "taesd3", "taef1"]
+    MAIN_CATEGORY = "Model Loading"
+
     @staticmethod
     def vae_list(s):
         vaes = folder_paths.get_filename_list("vae")
@@ -831,6 +842,8 @@ def load_vae(self, vae_name):
         return (vae,)
 
 class ControlNetLoader:
+    MAIN_CATEGORY = "Model Loading"
+
     @classmethod
     def INPUT_TYPES(s):
         return {"required": { "control_net_name": (folder_paths.get_filename_list("controlnet"), )}}
@@ -849,6 +862,8 @@ def load_controlnet(self, control_net_name):
         return (controlnet,)
 
 class DiffControlNetLoader:
+    MAIN_CATEGORY = "Model Loading"
+
     @classmethod
     def INPUT_TYPES(s):
         return {"required": { "model": ("MODEL",),
@@ -948,6 +963,8 @@ def apply_controlnet(self, positive, negative, control_net, image, strength, sta
 
 
 class UNETLoader:
+    MAIN_CATEGORY = "Model Loading"
+
     @classmethod
     def INPUT_TYPES(s):
         return {"required": { "unet_name": (folder_paths.get_filename_list("diffusion_models"), ),
@@ -973,6 +990,8 @@ def load_unet(self, unet_name, weight_dtype):
         return (model,)
 
 class CLIPLoader:
+    MAIN_CATEGORY = "Model Loading"
+
     @classmethod
     def INPUT_TYPES(s):
         return {"required": { "clip_name": (folder_paths.get_filename_list("text_encoders"), ),
@@ -1000,6 +1019,8 @@ def load_clip(self, clip_name, type="stable_diffusion", device="default"):
         return (clip,)
 
 class DualCLIPLoader:
+    MAIN_CATEGORY = "Model Loading"
+
     @classmethod
     def INPUT_TYPES(s):
         return {"required": { "clip_name1": (folder_paths.get_filename_list("text_encoders"), ),
@@ -1030,6 +1051,8 @@ def load_clip(self, clip_name1, clip_name2, type, device="default"):
         return (clip,)
 
 class CLIPVisionLoader:
+    MAIN_CATEGORY = "Model Loading"
+
     @classmethod
     def INPUT_TYPES(s):
         return {"required": { "clip_name": (folder_paths.get_filename_list("clip_vision"), ),
@@ -1066,6 +1089,8 @@ def encode(self, clip_vision, image, crop):
         return (output,)
 
 class StyleModelLoader:
+    MAIN_CATEGORY = "Model Loading"
+
     @classmethod
     def INPUT_TYPES(s):
         return {"required": { "style_model_name": (folder_paths.get_filename_list("style_models"), )}}
@@ -1165,6 +1190,8 @@ def apply_adm(self, conditioning, clip_vision_output, strength, noise_augmentati
         return (c, )
 
 class GLIGENLoader:
+    MAIN_CATEGORY = "Model Loading"
+
     @classmethod
     def INPUT_TYPES(s):
         return {"required": { "gligen_name": (folder_paths.get_filename_list("gligen"), )}}
diff --git a/server.py b/server.py
@@ -306,6 +306,37 @@
     "FloatToMask",
 ]}
 
+MAIN_CATEGORY_OVERRIDES.update({name: "Model Loading" for name in [
+    "IPAdapterInsightFaceLoader",
+    "IPAdapterModelLoader",
+    "IPAdapterUnifiedLoaderCommunity",
+    "ClownModelLoader",
+    "FluxLoader",
+    "SD35Loader",
+    "FluxGuidanceDisable",
+    "FluxOrthoCFGPatcher",
+    "ReAuraPatcher",
+    "ReAuraPatcherAdvanced",
+    "ReChromaPatcher",
+    "ReChromaPatcherAdvanced",
+    "ReFluxPatcher",
+    "ReFluxPatcherAdvanced",
+    "ReHiDreamPatcher",
+    "ReHiDreamPatcherAdvanced",
+    "ReLTXVPatcher",
+    "ReLTXVPatcherAdvanced",
+    "ReReduxPatcher",
+    "ReSD35Patcher",
+    "ReSD35PatcherAdvanced",
+    "ReSDPatcher",
+    "ReWanPatcher",
+    "ReWanPatcherAdvanced",
+    "TorchCompileModelAura",
+    "TorchCompileModelFluxAdv",
+    "TorchCompileModels",
+    "TorchCompileModelSD35",
+]})
+
 
 def _remove_sensitive_from_queue(queue: list) -> list:
     """Remove sensitive data (index 5) from queue item tuples."""
