diff --git a/backend/diffusion_engine/flux.py b/backend/diffusion_engine/flux.py index 0be2a7c1..fb9c2488 100644 --- a/backend/diffusion_engine/flux.py +++ b/backend/diffusion_engine/flux.py @@ -108,6 +108,7 @@ class Flux(ForgeDiffusionEngine): @torch.inference_mode() def decode_first_stage(self, x): + self.ref_latents.clear() sample = self.forge_objects.vae.first_stage_model.process_out(x) sample = self.forge_objects.vae.decode(sample).movedim(-1, 1) * 2.0 - 1.0 return sample.to(x) diff --git a/backend/diffusion_engine/qwen.py b/backend/diffusion_engine/qwen.py index 8241115b..42bddd15 100644 --- a/backend/diffusion_engine/qwen.py +++ b/backend/diffusion_engine/qwen.py @@ -107,6 +107,8 @@ class QwenImage(ForgeDiffusionEngine): @torch.inference_mode() def decode_first_stage(self, x): + self.ref_latents.clear() + self.image_prompt = "" sample = self.forge_objects.vae.first_stage_model.process_out(x) sample = self.forge_objects.vae.decode(sample).movedim(-1, 2) * 2.0 - 1.0 return sample.to(x)