small fix

2026-03-07 04:21:08 +00:00
parent 50037068f9
commit 1d898bac20
1 changed files with 1 additions and 1 deletions
@@ -724,7 +724,7 @@ class HeliosPipeline(DiffusionPipeline, HeliosLoraLoaderMixin):
                    )[0]

                if self.do_classifier_free_guidance:
-                    with self.transformer.cache_context("cond_uncond"):
+                    with self.transformer.cache_context("uncond"):
                        noise_uncond = self.transformer(
                            hidden_states=latents.to(transformer_dtype),
                            timestep=timestep,