In diffusion denoising model, we usually use classifier free guidance to control qualities. With State Manager, we want each context hold different state, so we have to set which context when calling self.transformer, here is a good example

But some popular pipelines (Flux2) or pipeline variants (Kontext, Image to image, Inpainting) don't implement this. For example, Flux T2I has it while Flux Kontext is missing

A Fix should be simple but require us to check all pipelines

with self.transformer.cache_context("cond"):
    noise_pred = self.transformer(...)
with self.transformer.cache_context("uncond"):
    neg_noise_pred = self.transformer(...)