From 9df7868dc77f836956231ab3a40ba18ba2b4be5a Mon Sep 17 00:00:00 2001 From: chrisgoringe Date: Fri, 27 Sep 2024 11:28:38 +1000 Subject: [PATCH] Make FluxTrainEnd an output node So that you don't need to use any of the outputs to get the thing to run! --- nodes.py | 1 + 1 file changed, 1 insertion(+) diff --git a/nodes.py b/nodes.py index 39d8d67..2076c9f 100644 --- a/nodes.py +++ b/nodes.py @@ -909,6 +909,7 @@ class FluxTrainEnd: RETURN_NAMES = ("lora_name", "metadata", "lora_path",) FUNCTION = "endtrain" CATEGORY = "FluxTrainer" + OUTPUT_NODE = True def endtrain(self, network_trainer, save_state): with torch.inference_mode(False):