fix: bug in sliced encoder

10 months ago · 8cfb46d6de
parent 372453e645
commit 8cfb46d6de
1 changed files with 1 additions and 1 deletions
--- a/imaginairy/modules/refiners_sd.py
+++ b/imaginairy/modules/refiners_sd.py
@ -127,7 +127,7 @@ class SD1AutoencoderSliced(SD1Autoencoder):
        """
        b, c, h, w = x.size()
        final_tensor = torch.zeros(
-            [1, 4, math.ceil(h / 8), math.ceil(w / 8)], device=x.device
+            [1, 4, math.floor(h / 8), math.floor(w / 8)], device=x.device
        )
        overlap_pct = 0.5