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