From e880d10951fcb05dda224738021e64d8dc84ef29 Mon Sep 17 00:00:00 2001 From: Haoming Date: Mon, 15 Dec 2025 11:50:48 +0800 Subject: [PATCH] clear --- backend/diffusion_engine/flux.py | 1 + backend/diffusion_engine/qwen.py | 2 ++ 2 files changed, 3 insertions(+) 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)