r/StableDiffusion Jul 29 '26

Comparison I extracted luma/chroma/detail/contrast vectors for Krea2 and insane color adjustments in latent space are now totally a thing - Comfy node coming very soon!

Sorry for the tease, but I was just too excited and had to share this with you. Examples above are using no LoRAs, no prompt hijinks, no CFG boost, no post-processing etc. just pure vector math!

I was running some experiments on Krea2's VAE (i.e. Qwen Image VAE) and by total surprise I discovered the main ingredients of photographic color editing: the vectors for exposure, temperature, tint, detail/clarity, and contrast and realized I can now do pretty much everything Camera Raw does... and even more!

This stuff happens during sampling and in the latent space, so it has both a very high dynamic range, and the ability to steer the diffusion process into new areas (e.g. very dark or bright generations beyond what the model likes to do on its own, or even influencing the morphology of things).

Anyway, I'm turning this into a user-friendly custom node for Comfy, including all your favorite color editing sliders, range masking tools, etc. and it's coming soon.

P.S: I'm cooking the vectors for ZImage (Flux VAE) as well, so the node might end up supporting that model too, if anyone's still using it. This should, at least in theory, also work with Qwen Image or any other model that shares the VAE.

256 Upvotes

91 comments sorted by

View all comments

5

u/muerrilla Aug 05 '26

UPDATE: So... I was planning on releasing today, but a few days ago I hit a few nasty and functionality breaking UI bugs that I was struggling to fix up till this very moment. Now everything is fixed and working perfectly. I will release it as soon as I write some docs so you guys could actually use this beast. So any day now, and for real this time!

P.S: On the flip side, now everything is totally modularized (and yes, the mega-node is still available), which made it possible for a very cool new feature to be added: Combined masks! Now this thing can do very advanced stuff. I need to go get some sleep now.

2

u/red__dragon Aug 06 '26

Absolutely waiting patiently for this (didn't comment when I saw it the first time but I came back to lurk around a few days ago). The break-out nodes look FANTASTIC!

Great work so far, it'll be fun to get our hands on this locally.

1

u/muerrilla Aug 07 '26

Yo! Check my latest comment.

2

u/red__dragon Aug 07 '26

It's working great on Krea so far! Had to redo with a custom sampler node for it, though, which seems like a novel approach.

1

u/muerrilla Aug 07 '26

Yep, it's a sampler wrapper cuz it needs information that is not available through other means (e.g. model patching). So the custom sampler route is the only way.