From fb5a1436c792eb8d115d9a0a7a6e02ac66ed83a6 Mon Sep 17 00:00:00 2001 From: nullquant Date: Thu, 2 May 2024 12:34:47 +0300 Subject: [PATCH] BlendInpaint inpaint resize to original --- brushnet_nodes.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/brushnet_nodes.py b/brushnet_nodes.py index 3b0423b..d5f0e6e 100644 --- a/brushnet_nodes.py +++ b/brushnet_nodes.py @@ -300,15 +300,15 @@ class BlendInpaint: blurred_mask = transform(mask[None,None,:,:]).to(original.device).to(original.dtype) inpaint = torch.nn.functional.interpolate( - inpaint, + inpaint.permute(0, 3, 1, 2), size=( - original.shape[1], - original.shape[2], + original.shape[0], + original.shape[1], ) ) ret = [] - for result in inpaint: + for result in inpaint.permute(0, 2, 3, 1): ret.append(original * (1.0 - blurred_mask[0][0][:,:,None]) + result.to(original.device) * blurred_mask[0][0][:,:,None]) return (torch.stack(ret), blurred_mask[0],)