From 3381a01c300c803567a595cb5a4f051f20ef5c0e Mon Sep 17 00:00:00 2001 From: TSC <112517630+LucianoCirino@users.noreply.github.com> Date: Thu, 15 Jun 2023 21:42:17 -0500 Subject: [PATCH] Added 'Output Only' option to KSampler --- efficiency_nodes.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/efficiency_nodes.py b/efficiency_nodes.py index 91dcd03..257354a 100644 --- a/efficiency_nodes.py +++ b/efficiency_nodes.py @@ -516,7 +516,7 @@ class TSC_KSampler: "negative": ("CONDITIONING",), "latent_image": ("LATENT",), "denoise": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.01}), - "preview_image": (["Disabled", "Enabled"],), + "preview_image": (["Disabled", "Enabled", "Output Only"],), }, "optional": { "optional_vae": ("VAE",), "script": ("SCRIPT",),}, @@ -695,10 +695,14 @@ class TSC_KSampler: results = preview_images(images, filename_prefix) update_value_by_id("results", my_unique_id, results) + # Determine what the 'images' value should be + images_value = list() if preview_image == "Output Only" else results + # Output image results to ui and node outputs - return {"ui": {"images": results}, + return {"ui": {"images": images_value}, "result": (model, positive, negative, {"samples": latent}, vae, images,)} + # ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ # If the sampler state is "Hold" elif sampler_state == "Hold": @@ -731,8 +735,11 @@ class TSC_KSampler: images = last_images results = last_results + # Determine what the 'images' value should be + images_value = list() if preview_image == "Output Only" else results + # Output image results to ui and node outputs - return {"ui": {"images": results}, + return {"ui": {"images": images_value}, "result": (model, positive, negative, {"samples": latent}, vae, images,)} # ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++