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

oops

Browse Source
This commit is contained in:
Haoming 2026-02-01 13:48:38 +08:00
parent cbcc2b51a6
commit 2edac94e68
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/processing.py
View File

@ -1467,12 +1467,14 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
self.sampler = sd_samplers.create_sampler(img2img_sampler_name, self.sd_model)
if self.latent_scale_mode is not None:
if _5d := (len(samples.shape) == 5):
samples = samples.squeeze(2)
for i in range(samples.shape[0]):
save_intermediate(samples, i)
if (_5d := len(samples.shape)) == 5:
samples = samples.squeeze(2)
samples = torch.nn.functional.interpolate(samples, size=(target_height // opt_f, target_width // opt_f), mode=self.latent_scale_mode["mode"], antialias=self.latent_scale_mode["antialias"])
if _5d:
samples = samples.unsqueeze(2)