Archives/imaginAIry
2
0
Fork 0
mirror of https://github.com/brycedrennan/imaginAIry synced 2024-11-19 03:25:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: bug in sliced encoder

Browse Source
This commit is contained in:
Bryce 2023-12-20 12:31:27 -08:00 committed by Bryce Drennan
parent 372453e645
commit 8cfb46d6de
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
imaginairy/modules/refiners_sd.py
View File

@ -127,7 +127,7 @@ class SD1AutoencoderSliced(SD1Autoencoder):
""" """
b, c, h, w = x.size() b, c, h, w = x.size()
final_tensor = torch.zeros( 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 overlap_pct = 0.5