From 7d979c0f06adfcfa964b24af1cb515008a5af688 Mon Sep 17 00:00:00 2001
From: Haoming <hmstudy02@gmail.com>
Date: Mon, 18 May 2026 09:52:49 +0800
Subject: [PATCH] dim

---
 modules/processing.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/modules/processing.py b/modules/processing.py
index 01db6ece..d5226482 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -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