From 148c805a15dbbf9657cc9f50a21efa5993db87c7 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Thu, 9 May 2024 19:14:05 +0300
Subject: [PATCH] Add ModelPassThrough -node

---
 __init__.py    |  1 +
 nodes/nodes.py | 21 +++++++++++++++++++++
 2 files changed, 22 insertions(+)

diff --git a/__init__.py b/__init__.py
index 3f248bd..4111484 100644
--- a/__init__.py
+++ b/__init__.py
@@ -90,6 +90,7 @@ NODE_CONFIG = {
     "VRAM_Debug": {"class": VRAM_Debug, "name": "VRAM Debug"},
     "SomethingToString": {"class": SomethingToString, "name": "Something To String"},
     "EmptyLatentImagePresets": {"class": EmptyLatentImagePresets, "name": "Empty Latent Image Presets"},
+    "ModelPassThrough": {"class": ModelPassThrough, "name": "ModelPass"},
     #audioscheduler stuff
     "NormalizedAmplitudeToMask": {"class": NormalizedAmplitudeToMask},
     "NormalizedAmplitudeToFloatList": {"class": NormalizedAmplitudeToFloatList},
diff --git a/nodes/nodes.py b/nodes/nodes.py
index ad40985..99f159f 100644
--- a/nodes/nodes.py
+++ b/nodes/nodes.py
@@ -280,6 +280,27 @@ class CondPassThrough:
     def passthrough(self, positive, negative):
         return (positive, negative,)
 
+class ModelPassThrough:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "model": ("MODEL", ),
+            }, 
+    }
+
+    RETURN_TYPES = ("MODEL", )
+    RETURN_NAMES = ("model",)
+    FUNCTION = "passthrough"
+    CATEGORY = "KJNodes/misc"
+    DESCRIPTION = """
+    Simply passes through the model,
+    workaround for Set node not allowing bypassed inputs.
+"""
+
+    def passthrough(self, model):
+        return (model,)
+
 def append_helper(t, mask, c, set_area_to_bounds, strength):
         n = [t[0], t[1].copy()]
         _, h, w = mask.shape