fix:run error when input optional image in easy fullkSampler #178
This commit is contained in:
+4
-1
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user