mirror of
https://github.com/vladmandic/automatic
synced 2026-09-18 16:54:33 +02:00
@@ -270,7 +270,7 @@ class TransparentVAEDecoder(AutoencoderKL):
|
||||
alpha = y[..., :1]
|
||||
fg = y[..., 1:]
|
||||
|
||||
B, H, W, C = fg.shape
|
||||
_B, H, W, _C = fg.shape
|
||||
cb = checkerboard(shape=(H // 64, W // 64))
|
||||
cb = cv2.resize(cb, (W, H), interpolation=cv2.INTER_LANCZOS4)
|
||||
cb = (0.5 + (cb - 0.5) * 0.1)[None, ..., None]
|
||||
|
||||
Reference in New Issue
Block a user