Make FluxTrainEnd an output node

So that you don't need to use any of the outputs to get the thing to run!
2024-09-27 11:28:38 +10:00
parent 4ad054cce5
commit 9df7868dc7
1 changed files with 1 additions and 0 deletions
@@ -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):