From 1d898bac20ea113c9fe2edcfcb37cb0de6967391 Mon Sep 17 00:00:00 2001 From: SHYuanBest Date: Sat, 7 Mar 2026 04:21:08 +0000 Subject: [PATCH] small fix --- helios/diffusers_version/pipeline_helios_diffusers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/helios/diffusers_version/pipeline_helios_diffusers.py b/helios/diffusers_version/pipeline_helios_diffusers.py index 37624f5..72e0a08 100644 --- a/helios/diffusers_version/pipeline_helios_diffusers.py +++ b/helios/diffusers_version/pipeline_helios_diffusers.py @@ -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,