From bfb5645d31461cd96329fb510796da587db3552a Mon Sep 17 00:00:00 2001
From: TenserTensor <vadim.oparin@icloud.com>
Date: Mon, 23 Feb 2026 09:11:15 +0300
Subject: [PATCH] Fixes: Workflow and loader nodes

---
 nodes/loaders/flux_models_loader_advanced.py       |  6 ++++++
 nodes/loaders/gguf_models_loader_advanced.py       |  6 ++++++
 nodes/workflow/flux2_workflow_settings_advanced.py | 11 +++++++----
 nodes/workflow/workflow_settings_helpers.py        |  2 +-
 4 files changed, 20 insertions(+), 5 deletions(-)

diff --git a/nodes/loaders/flux_models_loader_advanced.py b/nodes/loaders/flux_models_loader_advanced.py
index b55adfe..3ed2c8a 100644
--- a/nodes/loaders/flux_models_loader_advanced.py
+++ b/nodes/loaders/flux_models_loader_advanced.py
@@ -7,6 +7,12 @@ from .loader_helpers import load_unet, patch_flux_sampling, load_flux_clip, appl
 
 
 class TT_FluxModelsLoaderAdvanced():
+    def __init__(self):
+        self.loaded_lora_1 = None
+        self.loaded_lora_2 = None
+        self.loaded_lora_3 = None
+        self.loaded_lora_4 = None
+
     @classmethod
     def INPUT_TYPES(cls):
         UNET_DTYPES = ["default", "fp8_e4m3fn", "fp8_e4m3fn_fast", "fp8_e5m2"]
diff --git a/nodes/loaders/gguf_models_loader_advanced.py b/nodes/loaders/gguf_models_loader_advanced.py
index 4ba53d8..e2bd1c1 100644
--- a/nodes/loaders/gguf_models_loader_advanced.py
+++ b/nodes/loaders/gguf_models_loader_advanced.py
@@ -11,6 +11,12 @@ update_folder_names_and_paths("clip_gguf", ["text_encoders", "clip"])
 
 
 class TT_GgufModelsLoaderAdvanced():
+    def __init__(self):
+        self.loaded_lora_1 = None
+        self.loaded_lora_2 = None
+        self.loaded_lora_3 = None
+        self.loaded_lora_4 = None
+
     @classmethod
     def INPUT_TYPES(cls):
         UNET_NAMES = [x for x in FP.get_filename_list("unet_gguf")]
diff --git a/nodes/workflow/flux2_workflow_settings_advanced.py b/nodes/workflow/flux2_workflow_settings_advanced.py
index 96466d1..16eb7bc 100644
--- a/nodes/workflow/flux2_workflow_settings_advanced.py
+++ b/nodes/workflow/flux2_workflow_settings_advanced.py
@@ -28,11 +28,13 @@ class TT_Flux2WorkflowSettingsAdvanced:
         "INT",
         "INT",
         "FLOAT",
-        "SAMPLER",
         S.KSampler.SAMPLERS,
+        "SAMPLER",
         "SIGMAS",
         "INT",
         "INT",
+        "STRING",
+        "STRING",
         "FLOAT"
     )
     RETURN_NAMES = (
@@ -40,19 +42,20 @@ class TT_Flux2WorkflowSettingsAdvanced:
         "SEED",
         "STEPS",
         "CFG",
-        "SAMPLER",
         "SAMPLER_NAME",
+        "SAMPLER",
         "SCHEDULER",
         "WIDTH",
         "HEIGHT",
         "PROMPT",
+        "LORA_TRIGGERS",
         "GUIDANCE"
     )
     FUNCTION = "get_settings"
     CATEGORY = "TenserTensor/Workflow/FLUX2"
 
     def get_settings(self, seed, steps, cfg, sampler_name, width, height, prompt, lora_triggers, guidance):
-        scheduler = get_schedule(seed, width, height)
+        scheduler = get_schedule(steps, width, height)
         sampler = S.sampler_object(sampler_name)
 
         workflow_config = {
@@ -69,4 +72,4 @@ class TT_Flux2WorkflowSettingsAdvanced:
             "guidance": guidance
         }
 
-        return (workflow_config, seed, steps, cfg, sampler, scheduler, width, height, prompt, lora_triggers, guidance)
+        return (workflow_config, seed, steps, cfg, sampler_name, sampler, scheduler, width, height, prompt, lora_triggers, guidance)
diff --git a/nodes/workflow/workflow_settings_helpers.py b/nodes/workflow/workflow_settings_helpers.py
index 65f103b..e5e3fc5 100644
--- a/nodes/workflow/workflow_settings_helpers.py
+++ b/nodes/workflow/workflow_settings_helpers.py
@@ -33,4 +33,4 @@ def get_schedule(num_steps, width, height):
     timesteps = torch.linspace(1, 0, num_steps + 1)
     timesteps = _generalized_time_snr_shift(timesteps, mu, 1.0)
 
-    return (timesteps,)
+    return timesteps