feat: add Generation Setup main category mapping

Author: Yourz

comfy_extras/nodes_ace.py

@@ -12,6 +12,7 @@ def define_schema(cls):
         return io.Schema(
             node_id="TextEncodeAceStepAudio",
             category="conditioning",
+            main_category="Generation Setup",
             inputs=[
                 io.Clip.Input("clip"),
                 io.String.Input("tags", multiline=True, dynamic_prompts=True),
@@ -62,6 +63,7 @@ def define_schema(cls):
             node_id="EmptyAceStepLatentAudio",
             display_name="Empty Ace Step 1.0 Latent Audio",
             category="latent/audio",
+            main_category="Generation Setup",
             inputs=[
                 io.Float.Input("seconds", default=120.0, min=1.0, max=1000.0, step=0.1),
                 io.Int.Input(

comfy_extras/nodes_advanced_samplers.py

@@ -46,6 +46,7 @@ def define_schema(cls) -> io.Schema:
         return io.Schema(
             node_id="SamplerLCMUpscale",
             category="sampling/custom_sampling/samplers",
+            main_category="Generation Setup",
             inputs=[
                 io.Float.Input("scale_ratio", default=1.0, min=0.1, max=20.0, step=0.01),
                 io.Int.Input("scale_steps", default=-1, min=-1, max=1000, step=1),
@@ -93,6 +94,7 @@ def define_schema(cls) -> io.Schema:
             node_id="SamplerEulerCFGpp",
             display_name="SamplerEulerCFG++",
             category="_for_testing",  # "sampling/custom_sampling/samplers"
+            main_category="Generation Setup",
             inputs=[
                 io.Combo.Input("version", options=["regular", "alternative"]),
             ],

comfy_extras/nodes_align_your_steps.py

@@ -30,6 +30,7 @@ def define_schema(cls) -> io.Schema:
             node_id="AlignYourStepsScheduler",
             search_aliases=["AYS scheduler"],
             category="sampling/custom_sampling/schedulers",
+            main_category="Generation Setup",
             inputs=[
                 io.Combo.Input("model_type", options=["SD1", "SDXL", "SVD"]),
                 io.Int.Input("steps", default=10, min=1, max=10000),

comfy_extras/nodes_apg.py

@@ -17,6 +17,7 @@ def define_schema(cls) -> io.Schema:
             node_id="APG",
             display_name="Adaptive Projected Guidance",
             category="sampling/custom_sampling",
+            main_category="Generation Setup",
             inputs=[
                 io.Model.Input("model"),
                 io.Float.Input(

comfy_extras/nodes_audio.py

@@ -19,6 +19,7 @@ def define_schema(cls):
             node_id="EmptyLatentAudio",
             display_name="Empty Latent Audio",
             category="latent/audio",
+            main_category="Generation Setup",
             inputs=[
                 IO.Float.Input("seconds", default=47.6, min=1.0, max=1000.0, step=0.1),
                 IO.Int.Input(
@@ -43,6 +44,7 @@ def define_schema(cls):
         return IO.Schema(
             node_id="ConditioningStableAudio",
             category="conditioning",
+            main_category="Generation Setup",
             inputs=[
                 IO.Conditioning.Input("positive"),
                 IO.Conditioning.Input("negative"),
@@ -72,6 +74,7 @@ def define_schema(cls):
             search_aliases=["audio to latent"],
             display_name="VAE Encode Audio",
             category="latent/audio",
+            main_category="Generation Setup",
             inputs=[
                 IO.Audio.Input("audio"),
                 IO.Vae.Input("vae"),
@@ -102,6 +105,7 @@ def define_schema(cls):
             search_aliases=["latent to audio"],
             display_name="VAE Decode Audio",
             category="latent/audio",
+            main_category="Generation Setup",
             inputs=[
                 IO.Latent.Input("samples"),
                 IO.Vae.Input("vae"),

comfy_extras/nodes_audio_encoder.py

@@ -37,6 +37,7 @@ def define_schema(cls) -> io.Schema:
         return io.Schema(
             node_id="AudioEncoderEncode",
             category="conditioning",
+            main_category="Generation Setup",
             inputs=[
                 io.AudioEncoder.Input("audio_encoder"),
                 io.Audio.Input("audio"),

comfy_extras/nodes_chroma_radiance.py

@@ -14,6 +14,7 @@ def define_schema(cls) -> io.Schema:
         return io.Schema(
             node_id="EmptyChromaRadianceLatentImage",
             category="latent/chroma_radiance",
+            main_category="Generation Setup",
             inputs=[
                 io.Int.Input(id="width", default=1024, min=16, max=nodes.MAX_RESOLUTION, step=16),
                 io.Int.Input(id="height", default=1024, min=16, max=nodes.MAX_RESOLUTION, step=16),

comfy_extras/nodes_clip_sdxl.py

@@ -10,6 +10,7 @@ def define_schema(cls):
         return io.Schema(
             node_id="CLIPTextEncodeSDXLRefiner",
             category="advanced/conditioning",
+            main_category="Generation Setup",
             inputs=[
                 io.Float.Input("ascore", default=6.0, min=0.0, max=1000.0, step=0.01),
                 io.Int.Input("width", default=1024, min=0, max=nodes.MAX_RESOLUTION),
@@ -31,6 +32,7 @@ def define_schema(cls):
         return io.Schema(
             node_id="CLIPTextEncodeSDXL",
             category="advanced/conditioning",
+            main_category="Generation Setup",
             inputs=[
                 io.Clip.Input("clip"),
                 io.Int.Input("width", default=1024, min=0, max=nodes.MAX_RESOLUTION),

comfy_extras/nodes_cond.py

@@ -9,6 +9,7 @@ def define_schema(cls) -> io.Schema:
         return io.Schema(
             node_id="CLIPTextEncodeControlnet",
             category="_for_testing/conditioning",
+            main_category="Generation Setup",
             inputs=[
                 io.Clip.Input("clip"),
                 io.Conditioning.Input("conditioning"),
@@ -36,6 +37,7 @@ def define_schema(cls) -> io.Schema:
         return io.Schema(
             node_id="T5TokenizerOptions",
             category="_for_testing/conditioning",
+            main_category="Generation Setup",
             inputs=[
                 io.Clip.Input("clip"),
                 io.Int.Input("min_padding", default=0, min=0, max=10000, step=1),

comfy_extras/nodes_controlnet.py

@@ -10,6 +10,7 @@ def define_schema(cls):
         return io.Schema(
             node_id="SetUnionControlNetType",
             category="conditioning/controlnet",
+            main_category="Generation Setup",
             inputs=[
                 io.ControlNet.Input("control_net"),
                 io.Combo.Input("type", options=["auto"] + list(UNION_CONTROLNET_TYPES.keys())),
@@ -40,6 +41,7 @@ def define_schema(cls):
             node_id="ControlNetInpaintingAliMamaApply",
             search_aliases=["masked controlnet"],
             category="conditioning/controlnet",
+            main_category="Generation Setup",
             inputs=[
                 io.Conditioning.Input("positive"),
                 io.Conditioning.Input("negative"),