From 33566e84747b17b9a3b7a2c22c2dc28bb756fd8f Mon Sep 17 00:00:00 2001 From: yolain Date: Tue, 21 May 2024 15:41:39 +0800 Subject: [PATCH] fix:run error when input optional image in easy fullkSampler #178 --- py/easyNodes.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/py/easyNodes.py b/py/easyNodes.py index dc71a2c..18ea1cb 100644 --- a/py/easyNodes.py +++ b/py/easyNodes.py @@ -4197,7 +4197,7 @@ class samplerFull(LayerDiffuse): FUNCTION = "run" CATEGORY = "EasyUse/Sampler" - def run(self, pipe, steps, cfg, sampler_name, scheduler, denoise, image_output, link_id, save_prefix, seed=None, model=None, positive=None, negative=None, latent=None, vae=None, clip=None, xyPlot=None, tile_size=None, prompt=None, extra_pnginfo=None, my_unique_id=None, force_full_denoise=False, disable_noise=False, downscale_options=None): + def run(self, pipe, steps, cfg, sampler_name, scheduler, denoise, image_output, link_id, save_prefix, seed=None, model=None, positive=None, negative=None, latent=None, vae=None, clip=None, xyPlot=None, tile_size=None, prompt=None, extra_pnginfo=None, my_unique_id=None, force_full_denoise=False, disable_noise=False, downscale_options=None, image=None): samp_model = model.clone() if model is not None else pipe["model"].clone() samp_positive = positive if positive is not None else pipe["positive"] @@ -4220,6 +4220,9 @@ class samplerFull(LayerDiffuse): add_noise = pipe['loader_settings']['add_noise'] if 'add_noise' in pipe['loader_settings'] else 'enabled' force_full_denoise = pipe['loader_settings']['force_full_denoise'] if 'force_full_denoise' in pipe['loader_settings'] else True + if image is not None and latent is None: + samp_samples = {"samples": samp_vae.encode(image[:, :, :, :3])} + disable_noise = False if add_noise == "disable": disable_noise = True