Browse Source

Speed up Sharpen node.

pull/1690/merge
comfyanonymous 7 months ago
parent
commit
4ee9aad6ca
  1. 3
      comfy_extras/nodes_post_processing.py

3
comfy_extras/nodes_post_processing.py

@ -227,6 +227,7 @@ class Sharpen:
return (image,)
batch_size, height, width, channels = image.shape
image = image.to(comfy.model_management.get_torch_device())
kernel_size = sharpen_radius * 2 + 1
kernel = gaussian_kernel(kernel_size, sigma, device=image.device) * -(alpha*10)
@ -241,7 +242,7 @@ class Sharpen:
result = torch.clamp(sharpened, 0, 1)
return (result,)
return (result.to(comfy.model_management.intermediate_device()),)
class ImageScaleToTotalPixels:
upscale_methods = ["nearest-exact", "bilinear", "area", "bicubic", "lanczos"]

Loading…
Cancel
Save