From 60fc4f88e158704d929954fa8db943ee24a93f24 Mon Sep 17 00:00:00 2001 From: nullquant Date: Mon, 29 Apr 2024 11:27:22 +0300 Subject: [PATCH] blend_inpaint blurred_mask device patch --- brushnet_nodes.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/brushnet_nodes.py b/brushnet_nodes.py index 1f08a2d..70c528c 100644 --- a/brushnet_nodes.py +++ b/brushnet_nodes.py @@ -306,11 +306,11 @@ class BlendInpaint: if kernel % 2 == 0: kernel += 1 transform = T.GaussianBlur(kernel_size=(kernel, kernel), sigma=(sigma, sigma)) - blurred_mask = transform(mask[None,None,:,:]) + blurred_mask = transform(mask[None,None,:,:]).to(original.device).to(original.dtype) ret = [] for result in inpaint: - ret.append(original * (1.0 - blurred_mask[0][0][:,:,None]) + result * blurred_mask[0][0][:,:,None]) + 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],)