Merge branch 'main' of https://github.com/kijai/ComfyUI-FluxTrainer

2024-11-07 00:17:05 +02:00
parent 8bfe1f668b f05d2c3c96
commit a348bae0a2
1 changed files with 1 additions and 0 deletions
@@ -989,6 +989,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):