CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-21 21:01:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

clear

Browse Source
This commit is contained in:
Haoming 2025-12-15 11:50:48 +08:00
parent 48116663c0
commit e880d10951
2 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
backend/diffusion_engine/flux.py
View File

@ -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)

2
backend/diffusion_engine/qwen.py
View File

@ -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)