From 6a0c755c560baf89bbba08e2cd3eb51d01809b95 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Sun, 6 Jul 2025 23:06:20 +0300 Subject: [PATCH] these should not be output nodes... --- nodes/model_optimization_nodes.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/nodes/model_optimization_nodes.py b/nodes/model_optimization_nodes.py index e0b55fe..81a4252 100644 --- a/nodes/model_optimization_nodes.py +++ b/nodes/model_optimization_nodes.py @@ -179,7 +179,6 @@ class CheckpointLoaderKJ(BaseLoaderKJ): RETURN_TYPES = ("MODEL", "CLIP", "VAE") FUNCTION = "patch" - OUTPUT_NODE = True DESCRIPTION = "Experimental node for patching torch.nn.Linear with CublasLinear." EXPERIMENTAL = True CATEGORY = "KJNodes/experimental" @@ -321,7 +320,6 @@ class DiffusionModelLoaderKJ(BaseLoaderKJ): RETURN_TYPES = ("MODEL",) FUNCTION = "patch_and_load" - OUTPUT_NODE = True DESCRIPTION = "Node for patching torch.nn.Linear with CublasLinear." EXPERIMENTAL = True CATEGORY = "KJNodes/experimental" @@ -375,7 +373,6 @@ class ModelPatchTorchSettings: RETURN_TYPES = ("MODEL",) FUNCTION = "patch" - OUTPUT_NODE = True DESCRIPTION = "Adds callbacks to model to set torch settings before and after running the model." EXPERIMENTAL = True CATEGORY = "KJNodes/experimental" @@ -534,6 +531,7 @@ class TorchCompileModelFluxAdvanced: CATEGORY = "KJNodes/torchcompile" EXPERIMENTAL = True + DEPRECATED = True def parse_blocks(self, blocks_str): blocks = [] @@ -723,6 +721,7 @@ class TorchCompileModelWanVideo: CATEGORY = "KJNodes/torchcompile" EXPERIMENTAL = True + DEPCRECATED = True def patch(self, model, backend, fullgraph, mode, dynamic, dynamo_cache_size_limit, compile_transformer_blocks_only): m = model.clone()