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

dim

Browse Source
This commit is contained in:
Haoming 2026-05-18 09:52:49 +08:00
parent 3a5eafb7d3
commit 7d979c0f06
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/processing.py
View File

@ -1833,7 +1833,11 @@ class StableDiffusionProcessingImg2Img(StableDiffusionProcessing):
devices.torch_gc()
if self.resize_mode == 3:
if _5d := (self.init_latent.ndim == 5):
self.init_latent = self.init_latent.squeeze(2)
self.init_latent = torch.nn.functional.interpolate(self.init_latent, size=(self.height // opt_f, self.width // opt_f), mode="bilinear")
if _5d:
self.init_latent = self.init_latent.unsqueeze(2)
if image_mask is not None:
init_mask = latent_mask