|
|
|
@@ -1354,9 +1354,14 @@ def segs_to_masklist(segs):
|
|
|
|
|
return masks
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def vae_decode(vae, samples, use_tile, hook, tile_size=512):
|
|
|
|
|
def vae_decode(vae, samples, use_tile, hook, tile_size=512, overlap=64):
|
|
|
|
|
if use_tile:
|
|
|
|
|
pixels = nodes.VAEDecodeTiled().decode(vae, samples, tile_size)[0]
|
|
|
|
|
decoder = nodes.VAEDecodeTiled()
|
|
|
|
|
if 'overlap' in inspect.signature(decoder.decode).parameters:
|
|
|
|
|
pixels = decoder.decode(vae, samples, tile_size, overlap=overlap)[0]
|
|
|
|
|
else:
|
|
|
|
|
print(f"[Impact Pack] Your ComfyUI is outdated.")
|
|
|
|
|
pixels = decoder.decode(vae, samples, tile_size)[0]
|
|
|
|
|
else:
|
|
|
|
|
pixels = nodes.VAEDecode().decode(vae, samples)[0]
|
|
|
|
|
|
|
|
|
@@ -1378,12 +1383,12 @@ def vae_encode(vae, pixels, use_tile, hook, tile_size=512):
|
|
|
|
|
return samples
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def latent_upscale_on_pixel_space_shape(samples, scale_method, w, h, vae, use_tile=False, tile_size=512, save_temp_prefix=None, hook=None):
|
|
|
|
|
return latent_upscale_on_pixel_space_shape2(samples, scale_method, w, h, vae, use_tile, tile_size, save_temp_prefix, hook)[0]
|
|
|
|
|
def latent_upscale_on_pixel_space_shape(samples, scale_method, w, h, vae, use_tile=False, tile_size=512, save_temp_prefix=None, hook=None, overlap=64):
|
|
|
|
|
return latent_upscale_on_pixel_space_shape2(samples, scale_method, w, h, vae, use_tile, tile_size, save_temp_prefix, hook, overlap=overlap)[0]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def latent_upscale_on_pixel_space_shape2(samples, scale_method, w, h, vae, use_tile=False, tile_size=512, save_temp_prefix=None, hook=None):
|
|
|
|
|
pixels = vae_decode(vae, samples, use_tile, hook, tile_size=tile_size)
|
|
|
|
|
def latent_upscale_on_pixel_space_shape2(samples, scale_method, w, h, vae, use_tile=False, tile_size=512, save_temp_prefix=None, hook=None, overlap=64):
|
|
|
|
|
pixels = vae_decode(vae, samples, use_tile, hook, tile_size=tile_size, overlap=overlap)
|
|
|
|
|
|
|
|
|
|
if save_temp_prefix is not None:
|
|
|
|
|
nodes.PreviewImage().save_images(pixels, filename_prefix=save_temp_prefix)
|
|
|
|
@@ -1397,12 +1402,12 @@ def latent_upscale_on_pixel_space_shape2(samples, scale_method, w, h, vae, use_t
|
|
|
|
|
return (vae_encode(vae, pixels, use_tile, hook, tile_size=tile_size), old_pixels)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def latent_upscale_on_pixel_space(samples, scale_method, scale_factor, vae, use_tile=False, tile_size=512, save_temp_prefix=None, hook=None):
|
|
|
|
|
return latent_upscale_on_pixel_space2(samples, scale_method, scale_factor, vae, use_tile, tile_size, save_temp_prefix, hook)[0]
|
|
|
|
|
def latent_upscale_on_pixel_space(samples, scale_method, scale_factor, vae, use_tile=False, tile_size=512, save_temp_prefix=None, hook=None, overlap=64):
|
|
|
|
|
return latent_upscale_on_pixel_space2(samples, scale_method, scale_factor, vae, use_tile, tile_size, save_temp_prefix, hook, overlap=overlap)[0]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def latent_upscale_on_pixel_space2(samples, scale_method, scale_factor, vae, use_tile=False, tile_size=512, save_temp_prefix=None, hook=None):
|
|
|
|
|
pixels = vae_decode(vae, samples, use_tile, hook, tile_size=tile_size)
|
|
|
|
|
def latent_upscale_on_pixel_space2(samples, scale_method, scale_factor, vae, use_tile=False, tile_size=512, save_temp_prefix=None, hook=None, overlap=64):
|
|
|
|
|
pixels = vae_decode(vae, samples, use_tile, hook, tile_size=tile_size, overlap=overlap)
|
|
|
|
|
|
|
|
|
|
if save_temp_prefix is not None:
|
|
|
|
|
nodes.PreviewImage().save_images(pixels, filename_prefix=save_temp_prefix)
|
|
|
|
@@ -1418,12 +1423,12 @@ def latent_upscale_on_pixel_space2(samples, scale_method, scale_factor, vae, use
|
|
|
|
|
return (vae_encode(vae, pixels, use_tile, hook, tile_size=tile_size), old_pixels)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def latent_upscale_on_pixel_space_with_model_shape(samples, scale_method, upscale_model, new_w, new_h, vae, use_tile=False, tile_size=512, save_temp_prefix=None, hook=None):
|
|
|
|
|
return latent_upscale_on_pixel_space_with_model_shape2(samples, scale_method, upscale_model, new_w, new_h, vae, use_tile, tile_size, save_temp_prefix, hook)[0]
|
|
|
|
|
def latent_upscale_on_pixel_space_with_model_shape(samples, scale_method, upscale_model, new_w, new_h, vae, use_tile=False, tile_size=512, save_temp_prefix=None, hook=None, overlap=64):
|
|
|
|
|
return latent_upscale_on_pixel_space_with_model_shape2(samples, scale_method, upscale_model, new_w, new_h, vae, use_tile, tile_size, save_temp_prefix, hook, overlap=overlap)[0]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def latent_upscale_on_pixel_space_with_model_shape2(samples, scale_method, upscale_model, new_w, new_h, vae, use_tile=False, tile_size=512, save_temp_prefix=None, hook=None):
|
|
|
|
|
pixels = vae_decode(vae, samples, use_tile, hook, tile_size=tile_size)
|
|
|
|
|
def latent_upscale_on_pixel_space_with_model_shape2(samples, scale_method, upscale_model, new_w, new_h, vae, use_tile=False, tile_size=512, save_temp_prefix=None, hook=None, overlap=64):
|
|
|
|
|
pixels = vae_decode(vae, samples, use_tile, hook, tile_size=tile_size, overlap=overlap)
|
|
|
|
|
|
|
|
|
|
if save_temp_prefix is not None:
|
|
|
|
|
nodes.PreviewImage().save_images(pixels, filename_prefix=save_temp_prefix)
|
|
|
|
@@ -1450,12 +1455,12 @@ def latent_upscale_on_pixel_space_with_model_shape2(samples, scale_method, upsca
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def latent_upscale_on_pixel_space_with_model(samples, scale_method, upscale_model, scale_factor, vae, use_tile=False,
|
|
|
|
|
tile_size=512, save_temp_prefix=None, hook=None):
|
|
|
|
|
return latent_upscale_on_pixel_space_with_model2(samples, scale_method, upscale_model, scale_factor, vae, use_tile, tile_size, save_temp_prefix, hook)[0]
|
|
|
|
|
tile_size=512, save_temp_prefix=None, hook=None, overlap=64):
|
|
|
|
|
return latent_upscale_on_pixel_space_with_model2(samples, scale_method, upscale_model, scale_factor, vae, use_tile, tile_size, save_temp_prefix, hook, overlap=overlap)[0]
|
|
|
|
|
|
|
|
|
|
def latent_upscale_on_pixel_space_with_model2(samples, scale_method, upscale_model, scale_factor, vae, use_tile=False,
|
|
|
|
|
tile_size=512, save_temp_prefix=None, hook=None):
|
|
|
|
|
pixels = vae_decode(vae, samples, use_tile, hook, tile_size=tile_size)
|
|
|
|
|
tile_size=512, save_temp_prefix=None, hook=None, overlap=64):
|
|
|
|
|
pixels = vae_decode(vae, samples, use_tile, hook, tile_size=tile_size, overlap=overlap)
|
|
|
|
|
|
|
|
|
|
if save_temp_prefix is not None:
|
|
|
|
|
nodes.PreviewImage().save_images(pixels, filename_prefix=save_temp_prefix)
|
|
|
|
|