fix: bug in sliced encoder

This commit is contained in:
Bryce 2023-12-20 12:31:27 -08:00 committed by Bryce Drennan
parent 372453e645
commit 8cfb46d6de

View File

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