Image (Improved)

AugmentWaitImage - "Image (Improved)" in the menu - is the trigger-aware passthrough for the IMAGE type, part of Augment's Flow Control family. It's the node for img2img-style pipelines where one image has to travel through several stages and the stages have to actually run in the order you designed. Load → crop → place → refine → save; if any of those has ever come out stale because ComfyUI decided an intermediate didn't change, this is the node that fixes it.

The stock way to pass an image is just... a wire. And wires work - until caching kicks in and a stage skips because its inputs looked unchanged. The Wait node adds two things a plain wire can't: a trigger token that sequences the stages, and a guarantee (via the pack's executor patch) that it always re-runs instead of being pruned.

How it works

Read the source and you'll see it doesn't literally wait - execute passes the image straight through and emits a trigger token. The magic is in the pack's executor override: on import, Augment patches ComfyUI's execution internals so any node that takes or emits a TRIGGER type always re-runs and never gets cache-skipped as "unchanged." Thread trigger outputs into the next stage's trigger input and your image pipeline becomes a real ordered pipeline instead of a graph that fires everything at once.

So: it's a "make sure this image runs through here, in this order, every time" node. Nothing more, nothing less - and that's genuinely useful.

The inputs and outputs

• value (IMAGE, required) - the image to carry. Comes in on the wire; no widget on this one.
• trigger (optional) - the sequencing input from a previous stage.
• Outputs value (IMAGE), json (AUGMENT_JSON), trigger (TRIGGER) - the image passes through pixel-for-pixel unchanged, the trigger continues the chain, and json carries a structured dump (type and value as a string) for Augment's JSON Extract / JSON Viewer.

Nothing to configure. If you've got an image that must reach a specific sampler or save node in a specific order, this is your carrier.

How to install

Augment isn't on the ComfyUI registry yet, so it's a manual clone. The folder name is the classic gotcha:

cd ComfyUI/custom_nodes
git clone https://github.com/augment-lib/augment-ComfyUI augment
cd augment
pip install -r requirements.txt

The clone target must be augment - the pack imports everything as custom_nodes.augment.py.*, so the default folder name breaks loading. Restart ComfyUI afterward; nodes appear under Augment.

Dependencies are light (opencv-python, numpy, torch, Pillow, requests) and you already have them from ComfyUI. No model files, no API key - the paid API only applies to the pack's vector nodes.

Common issues

• Stages feeding stale images - cache behavior. Threading the image through this node and sequencing triggers makes each stage re-run.
• Node not showing after install - folder name. Rename to augment and restart.
• Everything re-executes now - by design. Trigger nodes always re-run, so use Wait nodes where ordering matters, not on every wire.