Skip to content

Commit 388150d

Browse files
authored
make kernel of same type as image to avoid mismatch issues (comfyanonymous#9932)
1 parent 20f665b commit 388150d

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

comfy_extras/nodes_post_processing.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -233,6 +233,7 @@ def sharpen(self, image: torch.Tensor, sharpen_radius: int, sigma:float, alpha:
233233

234234
kernel_size = sharpen_radius * 2 + 1
235235
kernel = gaussian_kernel(kernel_size, sigma, device=image.device) * -(alpha*10)
236+
kernel = kernel.to(dtype=image.dtype)
236237
center = kernel_size // 2
237238
kernel[center, center] = kernel[center, center] - kernel.sum() + 1.0
238239
kernel = kernel.repeat(channels, 1, 1).unsqueeze(1)

0 commit comments

Comments
 (0)