|
|
@ -768,6 +768,25 @@ class LatentUpscale: |
|
|
|
s["samples"] = comfy.utils.common_upscale(samples["samples"], width // 8, height // 8, upscale_method, crop) |
|
|
|
s["samples"] = comfy.utils.common_upscale(samples["samples"], width // 8, height // 8, upscale_method, crop) |
|
|
|
return (s,) |
|
|
|
return (s,) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class LatentUpscaleBy: |
|
|
|
|
|
|
|
upscale_methods = ["nearest-exact", "bilinear", "area", "bislerp"] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@classmethod |
|
|
|
|
|
|
|
def INPUT_TYPES(s): |
|
|
|
|
|
|
|
return {"required": { "samples": ("LATENT",), "upscale_method": (s.upscale_methods,), |
|
|
|
|
|
|
|
"scale_by": ("FLOAT", {"default": 1.5, "min": 0.01, "max": 8.0, "step": 0.01}),}} |
|
|
|
|
|
|
|
RETURN_TYPES = ("LATENT",) |
|
|
|
|
|
|
|
FUNCTION = "upscale" |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
CATEGORY = "latent" |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def upscale(self, samples, upscale_method, scale_by): |
|
|
|
|
|
|
|
s = samples.copy() |
|
|
|
|
|
|
|
width = round(samples["samples"].shape[3] * scale_by) |
|
|
|
|
|
|
|
height = round(samples["samples"].shape[2] * scale_by) |
|
|
|
|
|
|
|
s["samples"] = comfy.utils.common_upscale(samples["samples"], width, height, upscale_method, "disabled") |
|
|
|
|
|
|
|
return (s,) |
|
|
|
|
|
|
|
|
|
|
|
class LatentRotate: |
|
|
|
class LatentRotate: |
|
|
|
@classmethod |
|
|
|
@classmethod |
|
|
|
def INPUT_TYPES(s): |
|
|
|
def INPUT_TYPES(s): |
|
|
@ -1244,6 +1263,7 @@ NODE_CLASS_MAPPINGS = { |
|
|
|
"VAELoader": VAELoader, |
|
|
|
"VAELoader": VAELoader, |
|
|
|
"EmptyLatentImage": EmptyLatentImage, |
|
|
|
"EmptyLatentImage": EmptyLatentImage, |
|
|
|
"LatentUpscale": LatentUpscale, |
|
|
|
"LatentUpscale": LatentUpscale, |
|
|
|
|
|
|
|
"LatentUpscaleBy": LatentUpscaleBy, |
|
|
|
"LatentFromBatch": LatentFromBatch, |
|
|
|
"LatentFromBatch": LatentFromBatch, |
|
|
|
"RepeatLatentBatch": RepeatLatentBatch, |
|
|
|
"RepeatLatentBatch": RepeatLatentBatch, |
|
|
|
"SaveImage": SaveImage, |
|
|
|
"SaveImage": SaveImage, |
|
|
@ -1322,6 +1342,7 @@ NODE_DISPLAY_NAME_MAPPINGS = { |
|
|
|
"LatentCrop": "Crop Latent", |
|
|
|
"LatentCrop": "Crop Latent", |
|
|
|
"EmptyLatentImage": "Empty Latent Image", |
|
|
|
"EmptyLatentImage": "Empty Latent Image", |
|
|
|
"LatentUpscale": "Upscale Latent", |
|
|
|
"LatentUpscale": "Upscale Latent", |
|
|
|
|
|
|
|
"LatentUpscaleBy": "Upscale Latent By", |
|
|
|
"LatentComposite": "Latent Composite", |
|
|
|
"LatentComposite": "Latent Composite", |
|
|
|
"LatentFromBatch" : "Latent From Batch", |
|
|
|
"LatentFromBatch" : "Latent From Batch", |
|
|
|
"RepeatLatentBatch": "Repeat Latent Batch", |
|
|
|
"RepeatLatentBatch": "Repeat Latent Batch", |
|
|
|