From f1df6d15973fd765ef1868d189b2010434e1312b Mon Sep 17 00:00:00 2001 From: Haoming Date: Mon, 20 Oct 2025 12:03:46 +0800 Subject: [PATCH] bruh --- backend/diffusion_engine/qwen.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/backend/diffusion_engine/qwen.py b/backend/diffusion_engine/qwen.py index 41f25f10..8a9f2451 100644 --- a/backend/diffusion_engine/qwen.py +++ b/backend/diffusion_engine/qwen.py @@ -86,7 +86,8 @@ class QwenImage(ForgeDiffusionEngine): height = round(samples.shape[2] * scale_by / 8.0) * 8 s = torch.nn.functional.interpolate(samples, size=(height, width), mode="area") - self.ref_latents.append(self.forge_objects.vae.encode(s.movedim(1, -1)[:, :, :, :3])) + sample = self.forge_objects.vae.encode(s.movedim(1, -1)[:, :, :, :3]) + self.ref_latents.append(self.forge_objects.vae.first_stage_model.process_in(sample)) self.image_prompt += f"Picture {len(self.images_vl)}: <|vision_start|><|image_pad|><|vision_end|>"