Skip to content

rgthree/rgthree-comfy

Repository files navigation

rgthree-comfy
Making ComfyUI more comfortable!

The Nodes   |   Improvements & Features   |   Link Fixer


A collection of nodes and improvements created while messing around with ComfyUI. I made them for myself to make my workflow cleaner, easier, and faster. You're welcome to try them out. But remember, I made them for my own use cases :)

Context Node

Get Started

Install

  1. Install the great ComfyUi.
  2. Clone this repo into custom_modules:
    cd ComfyUI/custom_nodes
    git clone https://github.com/rgthree/rgthree-comfy.git
    
  3. Start up ComfyUI.

Settings

You can configure certain aspect of rgthree-comfy. For instance, perhaps a future ComfyUI change breaks rgthree-comfy, or you already have another extension that does something similar and you want to turn it off for rgthree-comfy.

You can get to rgthree-settings by right-clicking on the empty part of the graph, and selecting rgthree-comfy > Settings (rgthree-comfy) or by clicking the rgthree-comfy settings in the ComfyUI settings dialog.

(Note, settings are stored in an rgthree_config.json in the rgthree-comfy directory. There are other advanced settings that can only be configured there; You can copy default settings from rgthree_config.json.default before rgthree_config.json before modifying).


✴️ The Nodes

Note, you can right-click on a bunch of the rgthree-comfy nodes and select 🛟 Node Help menu item for in-app help when available.

Seed

An intuitive seed control node for ComfyUI that works very much like Automatic1111's seed control.

ℹ️ See More Information
  • Set the seed value to "-1" to use a random seed every time
  • Set any other number in there to use as a static/fixed seed
  • Quick actions to randomize, or (re-)use the last queued seed.
  • Images metadata will store the seed value (so dragging an image in, will have the seed field already fixed to its seed).
  • Secret Features: You can manually set the seed value to "-2" or "-3" to increment or decrement the last seed value. If there was not last seed value, it will randomly use on first.

Router Node

Reroute

Keep your workflow neat with this much improved Reroute node with, like, actual rerouting with multiple directions and sizes.

ℹ️ More Information
  • Use the right-click context menu to change the width, height and connection layout
  • Also toggle resizability (min size is 40x43 if resizing though), and title/type display.

Router Node

Bookmark (🔖)

Place the bookmark node anywhere on screen to quickly navigate to that with a shortcut key.

ℹ️ See More Information
  • Define the sahortvut_key to press to go right to that bookmark node, anchored in the top left
  • You can also define the zoom level as well!

Context / Context Big

Pass along in general flow properties, and merge in new data. Similar to some other node suites "pipes" but easier merging, is more easily interoperable with standard nodes by both combining and exploding all in a single node.

ℹ️ More Information
  • Context and Context Big are backwards compatible with each other. That is, an input connected to a Context Big will be passed through the CONTEXT outputs through normal Context nodes and available as an output on either (or, Context Big if the output is only on that node, like "steps").
  • Pro Tip: When dragging a Context output over a nother node, hold down "ctrl" and release to automatically connect the other Context outputs to the hovered node.
  • Pro Tip: You can change between Context and Context Big nodes from the menu.

Context Node

Image Comparer

The Image Comparer node compares two images on top of each other.

ℹ️ More Information
  • Note: The right-click menu may show image options (Open Image, Save Image, etc.) which will correspond to the first image (image_a) if clicked on the left-half of the node, or the second image if on the right half of the node.
  • Inputs:
    • image_a Required. The first image to use to compare. If image_b is not supplied and image_a is a batch, the comparer will use the first two images of image_a.
    • image_b Optional. The second image to use to compare. Optional only if image_a is a batch with two images.
  • Properties: You can change the following properties (by right-clicking on the node, and select "Properties" or "Properties Panel" from the menu):
    • comparer_mode - Choose between "Slide" and "Click". Defaults to "Slide".

Image Inset Crop

The node that lets you crop an input image by either pixel value, or percentage value.

Display Any

Displays most any piece of text data from the backend after execution.

Power Lora Loader

A super-simply Lora Loader node that can load multiple Loras at once, and quick toggle each, all in an ultra-condensed node.

ℹ️ More Information
  • Add as many Lora's as you would like by clicking the "+ Add Lora" button. There's no real limit!
  • Right-click on a Lora widget for special options to move the lora up or down (no affect on image, just presentation), toggle it on/off, or delete the row all together.
  • from the properties, change the Show Strengths to choose between showing a single, simple strength value (which will be used for both model and clip), or a more advanced view with both model and clip strengths being modifiable.

Lora Loader Stack

Deprecated. Used the Power Lora Loader instead.

A simplified Lora Loader stack. Much like other suites, but more interoperable with standard inputs/outputs.

Power Prompt

Power up your prompt and get drop downs for adding your embeddings, loras, and even have saved prompt snippets.

ℹ️ More Information
  • At the core, you can use Power Prompt almost as a String Primitive node with additional features of dropdowns for choosing your embeddings, and even loras, with no further processing. This will output just the raw TEXT to another node for any lora processing, CLIP Encoding, etc.
  • Connect a CLIP to the input to encode the text, with both the CLIP and CONDITIONING output right from the node.
  • Connect a MODEL to the input to parse and load any <lora:...> tags in the text automatically, without needing a separate Lora Loaders

Power Prompt - Simple

Same as Power Prompt above, but without LORA support; made for a slightly cleaner negative prompt (since negative prompts do not support loras).

SDXL Power Prompt - Positive

The SDXL sibling to the Power Prompt above. It contains the text_g and text_l as separate text inputs, as well a couple more input slots necessary to ensure proper clipe encoding. Combine with

SDXL Power Prompt - Simple

Like the non-SDXL Power Prompt - Simple node, this one is essentially the same as the SDXL Power Prompt but without lora support for either non-lora positive prompts or SDXL negative prompts (since negative prompts do not support loras).

SDXL Config

Just some configuration fields for SDXL prompting. Honestly, could be used for non SDXL too.

Context Switch / Context Switch Big

A powerful node to branch your workflow. Works by choosing the first Context input that is not null/empty.

ℹ️ More Information
  • Pass in several context nodes and the Context Switch will automatically choose the first non-null context to continue onward with.
  • Wondering how to toggle contexts to null? Use in conjuction with the Fast Muter or Fast Groups Muter

Any Switch

A powerful node to similar to the Context Switch above, that chooses the first input that is not null/empty.

ℹ️ More Information
  • Pass in several inmputs of the same type and the Any Switch will automatically choose the first non-null value to continue onward with.
  • Wondering how to toggle contexts to null? Use in conjuction with the Fast Muter or Fast Groups Muter

Fast Groups Muter

The Fast Groups Muter is an input-less node that automatically collects all groups in your current workflow and allows you to quickly mute and unmute all nodes within the group.

ℹ️ More Information
  • Groups will automatically be shown, though you can filter, sort and more from the node Properties (by right-clicking on the node, and select "Properties" or "Properties Panel" from the menu). Properties include:

  • matchColors - Only add groups that match the provided colors. Can be ComfyUI colors (red, pale_blue) or hex codes (#a4d399). Multiple can be added, comma delimited.

  • matchTitle - Filter the list of toggles by title match (string match, or regular expression).

  • showNav - Add / remove a quick navigation arrow to take you to the group. (default: true)

  • sort - Sort the toggles' order by "alphanumeric", graph "position", or "custom alphabet". (default: "position")

  • customSortAlphabet - When the sort property is "custom alphabet" you can define the alphabet to use here, which will match the beginning of each group name and sort against it. If group titles do not match any custom alphabet entry, then they will be put after groups that do, ordered alphanumerically.

    This can be a list of single characters, like "zyxw..." or comma delimited strings for more control, like "sdxl,pro,sd,n,p".

    Note, when two group title match the same custom alphabet entry, the normal alphanumeric alphabet breaks the tie. For instance, a custom alphabet of "e,s,d" will order groups names like "SDXL, SEGS, Detailer" eventhough the custom alphabet has an "e" before "d" (where one may expect "SE" to be before "SD").

    To have "SEGS" appear before "SDXL" you can use longer strings. For instance, the custom alphabet value of "se,s,f" would work here.

  • toggleRestriction - Optionally, attempt to restrict the number of widgets that can be enabled to a maximum of one, or always one.

    Note: If using "max one" or "always one" then this is only enforced when clicking a toggle on this node; if nodes within groups are changed outside of the initial toggle click, then these restriction will not be enforced, and could result in a state where more than one toggle is enabled. This could also happen if nodes are overlapped with multiple groups.

Fast Groups Bypasser

Same as Fast Groups Muter above, but sets the connected nodes to "Bypass" instead of "Mute"

Fast Muter

A powerful 'control panel' node to quickly toggle connected nodes allowing them to quickly be muted or enabled

ℹ️ More Information
  • Add a collection of all connected nodes allowing a single-spot as a "dashboard" to quickly enable and disable nodes. Two distinct nodes; one for "Muting" connected nodes, and one for "Bypassing" connected nodes.

Fast Bypasser

Same as Fast Muter but sets the connected nodes to "Bypass"

Fast Actions Button

Oh boy, this node allows you to semi-automate connected nodes and/or ConfyUI.

ℹ️ More Information
  • Connect nodes and, at the least, mute, bypass or enable them when the button is pressed.
  • Certain nodes expose additional actions. For instance, the Seed node you can set Randomize Each Time or Use Last Queued Seed when the button is pressed.
  • Also, from the node properties, set a shortcut key to toggle the button actions, without needing a click!

Node Collector

Used to cleanup noodles, this will accept any number of input nodes and passes it along to another node.

⚠️ Currently, this should really only be connected to Fast Muter, Fast Bypasser, or Mute / Bypass Relay.

Mute / Bypass Repeater

A powerful node that will dispatch its Mute/Bypass/Active mode to all connected input nodes or, if in a group w/o any connected inputs, will dispatch its Mute/Bypass/Active mode to all nodes in that group.

ℹ️ More Information
  • 💡 Pro Tip #1: Connect this node's output to a Fast Muter or Fast Bypasser to have a single toggle there that can mute/bypass/enable many nodes with one click.

  • 💡 Pro Tip #2: Connect a Mute / Bypass Relay node to this node's inputs to have the relay automatically dispatch a mute/bypass/enable change to the repeater.

Mute / Bypass Relay

An advanced node that, when working with a Mute / Bypass Repeater will relay a mute/bypass/activate signal to the repeater

ℹ️ More Information - Useful when you want a specific node or set of nodes to be muted when a different set of nodes are also muted.

Random Unmuter

An advanced node used to unmute one of its inputs randomly when the graph is queued (and, immediately mute it back).

ℹ️ More Information
  • Note: All input nodes MUST be muted to start; if not this node will not randomly unmute another. (This is powerful, as the generated image can be dragged in and the chosen input will already by unmuted and work w/o any further action.)
  • Tip: Connect a Repeater's output to this nodes input and place that Repeater on a group without any other inputs, and it will mute/unmute the entire group.

Advanced Techniques

First, a word on muting

A lot of the power of these nodes comes from Muting. Muting is the basis of correctly implementing multiple paths for a workflow utlizing the Context Switch node.

While other extensions may provide switches, they often get it wrong causing your workflow to do more work than is needed. While other switches may have a selector to choose which input to pass along, they don't stop the execution of the other inputs, which will result in wasted work. Instead, Context Switch works by choosing the first non-empty context to pass along and correctly Muting is one way to make a previous node empty, and causes no extra work to be done when set up correctly.

To understand muting, is to understand the graph flow

Muting, and therefore using Switches, can often confuse people at first because it feels like muting a node, or using a switch, should be able to stop or direct the forward flow of the graph. However, this is not the case and, in fact, the graph actually starts working backwards.

If you have a workflow that has a path like ... > Context > KSampler > VAE Decode > Save Image it may initially feel like you should be able to mute that first Context node and the graph would stop there when moving forward and skip the rest of that workflow.

But you'll quickly find that will cause an error, becase the graph doesn't actually move forward. When a workflow is processed, it first moves backwards starting at each "Output Node" (Preview Image, Save Image, even "Display String" etc.) and then walking backwards to all possible paths to get there.

So, with that ... > Context > KSampler > VAE Decode > Save Image example from above, we actually want to mute the Save Image node to stop this path. Once we do, since the output node is gone, none of these nodes will be run.

Let's take a look at an example.

A powerful combination: Using Context, Context Switch, & Fast Muter

Context Node

  1. Using the Context Switch (aqua colored in screenshot) feed context inputs in order of preference. In the workflow above, the Upscale Out context is first so, if that one is enabled, it will be chosen for the output. If not, the second input slot which comes from the context rerouted from above (before the Upscaler booth) will be chosen.

    • Notice the Upscale Preview is after the Upscale Out context node, using the image from it instead of the image from the upscale VAE Decoder. This is on purpose so, when we disable the Upscale Out context, none of the Upscaler nodes will run, saving precious GPU cycles. If we had the preview hooked up directly to the VAE Decoder the upscaler would always run to generate the preview, even if we had the Upscale Out context node disabled.
  2. We can now disable the Upscale Out context node by muting it. Highlighting it and pressing ctrl + m will work. By doing so, it's output will be None, and it will not pass anthing onto the further nodes. In the diagram you can see the Upscale Preview is red, but that's OK; there are no actual errors to stop execution.

  3. Now, let's hook it up to the Fast Muter node. The Fast Muter node works as dashboard by adding quick toggles for any connected node (ignoring reroutes). In the diagram, we have both the Upscaler Out context node, and the Save File context node hooked up. So, we can quickly enable and disable those.

    • The workflow seen here would be a common one where we can generate a handful of base previews cheaply with a random seed, and then choose one to upscale and save to disk.
  4. Lastly, and optionally, you can see the Node Collector. Use it to clean up noodles if you want and connect it to the muter. You can connect anything to it, but doing so may break your workflow's execution.


⚡ Improvements & Features

rgthree-comfy adds several improvements, features, and optimizations to ComfyUI that are not directly tied to nodes.

Progress Bar

A minimal progress bar that run alongs the top of the app window that shows the queue size, the current progress of the a prompt execution (within the same window), and the progress of multi-step nodes as well.

You can remove/enable from rgthree-comfy settings, as well as configure the height/size.

ComfyUI Recursive Optimization

An optimization to ComfyUI's recursive execution. Because rgthree-comfy nodes make it easy to build larger, more complex workflows, I (and others) started to hit a wall of poor execution times.

ℹ️ More Information
  • Until ComfyUI/issues/1502 is resolved and/or ComfyUI/pull/1503 is pulled in, then know that you're benefiting from hundreds of millions of saved cycles each run.

  • Specifically, for a rather complex test workflow, the patch reduces iterations of recursive_will_execute from 113,292,566 to just 135 (and 116.32 seconds to 69.84 seconds on my machine) on a fresh queue, and reduces recursive calls of recursive_output_delete_if_changed from 250,496,808 to 142 (and 158.13 seconds to 0.0 seconds on my machine).

  • ⚠️ However, there is a chance ComfyUI changes something in/around the code I patched which could break. If that's the case, you should disable the optimization from rgthree-comfy settings.

"Queue Selected Output Nodes" in right-click menu

Sometimes you want to just queue one or two paths to specific output node(s) without executing the entire workflow. Well, now you can do just that by right-clicking on an output node and selecting Queue Selected Output Nodes (rgthree).

ℹ️ More Information
  • Select the output nodes you want to execute.

  • Note: Only output nodes are captured and traversed, not all selected nodes. So if you select an output AND a node from a different path, only the path connected to the output will be executed and not non-output nodes, even if they were selected.

  • Note: The whole workflow is serialized, and then we trim what we don't want for the backend. So things like all seed random/increment/decrement will run even if that node isn't being sent in the end, etc.

Auto-Nest Subdirectories in long Combos

(Off by default while experimenting, turn on in rgthree-comfy settings).

Automatically detect top-level subdirectories in long combo lists (like, Load Checkpoint) and break out into sub directories.

Quick Mute/Bypass Toggles in Group Headers

(Off by default while experimenting, turn on in rgthree-comfy settings).

Adds a mute and/or bypass toggle icons in the top-right of Group Headers for one-click toggling of groups you may be currently looking at.

"Copy Image" in right-click menu

Right clicking on a node that has an image should have a context-menu item of "Copy Image" will allow you to copy the image right to your clipboard

🎓 I believe this has graduated, with ComfyUI recently adding this setting too. You won't get two menu items; my code checks that there isn't already a "Copy Image" item there before adding it.

Other/Smaller Fixes

  • Fixed the width of ultra-wide node chooser on double click.
  • Fixed z-indexes for textareas that would overlap above other elements, like Properties Panel, or @pythongosssss's image viewer.
  • Check for bad links when loading a workflow and log to console, by default. (See Link Fixer below).

📄 Link Fixer

If your workflows sometimes have missing connections, or even errors on load, start up ComfyUI and go to http://127.0.0.1:8188/rgthree/link_fixer which will allow you to drop in an image or workflow json file and check for and fix any bad links.

You can also enable a link fixer check in the rgthree-comfy settings to give you an alert if you load a workflow with bad linking data to start.