Fixes: Workflow and loader nodes

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"]

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")]

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)

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