Upload folder using huggingface_hub

README.md

@@ -1,5 +1,5 @@
 ---
-title: ImageGen - FLUX.2
+title: ImageGen - FLUX.2-KV
 emoji: 🖼
 colorFrom: purple
 colorTo: red

app.py

@@ -1,7 +1,6 @@
 import spaces
 import os
 import sys
-import requests
 import site
 
 APP_DIR = os.path.dirname(os.path.abspath(__file__))
@@ -45,106 +44,14 @@ def dummy_gpu_for_startup():
     print("--- [GPU Startup] Startup check passed. ---")
     return "Startup check passed."
 
-def handle_private_downloads():
-    """
-    Checks for a private_file_list.yaml, downloads required models using HF_TOKEN,
-    and then clears the token from the environment.
-    """
-    import yaml
-    from huggingface_hub import hf_hub_download
-    from core.settings import (
-        DIFFUSION_MODELS_DIR, TEXT_ENCODERS_DIR, VAE_DIR, CHECKPOINT_DIR, 
-        LORA_DIR, CONTROLNET_DIR, MODEL_PATCHES_DIR, EMBEDDING_DIR
-    )
-
-    print("--- [Startup] Checking for private models to download... ---")
-    private_list_path = os.path.join(APP_DIR, 'yaml', 'private_file_list.yaml')
-
-    if not os.path.exists(private_list_path):
-        print("--- [Startup] No private model list found. Skipping. ---")
-        if 'HF_TOKEN' in os.environ:
-             del os.environ['HF_TOKEN']
-             print("--- [Startup] Cleared HF_TOKEN environment variable as it is no longer needed. ---")
-             print(f"--- [Startup] Verifying HF_TOKEN after clearing: {os.environ.get('HF_TOKEN')}")
-        return
-
-    try:
-        with open(private_list_path, 'r', encoding='utf-8') as f:
-            private_files_config = yaml.safe_load(f)
-
-        if not private_files_config or 'file' not in private_files_config:
-            print("--- [Startup] Private model list is empty or malformed. Skipping. ---")
-            return
-        
-        category_to_dir_map = {
-            "diffusion_models": DIFFUSION_MODELS_DIR,
-            "text_encoders": TEXT_ENCODERS_DIR,
-            "vae": VAE_DIR,
-            "checkpoints": CHECKPOINT_DIR,
-            "loras": LORA_DIR,
-            "controlnet": CONTROLNET_DIR,
-            "model_patches": MODEL_PATCHES_DIR,
-            "embeddings": EMBEDDING_DIR,
-        }
-
-        files_to_download = []
-        for category, files in private_files_config.get('file', {}).items():
-            dest_dir = category_to_dir_map.get(category)
-            if not dest_dir:
-                print(f"--- [Startup] ⚠️ Unknown category '{category}' in private_file_list.yaml. Skipping. ---")
-                continue
-            
-            if isinstance(files, list):
-                for file_info in files:
-                    files_to_download.append((file_info, dest_dir))
-
-        if not files_to_download:
-            print("--- [Startup] No private models configured for download. ---")
-            return
-
-        print(f"--- [Startup] Found {len(files_to_download)} private model(s) to download. Using HF_TOKEN if available. ---")
-
-        for file_info, dest_dir in files_to_download:
-            filename = file_info.get("filename")
-            repo_id = file_info.get("repo_id")
-            repo_path = file_info.get("repository_file_path", filename)
-            
-            if not all([filename, repo_id]):
-                print(f"--- [Startup] ⚠️ Skipping malformed entry in private_file_list.yaml: {file_info} ---")
-                continue
-
-            dest_path = os.path.join(dest_dir, filename)
-            if os.path.lexists(dest_path):
-                print(f"--- [Startup] ✅ Model '{filename}' already exists. Skipping download. ---")
-                continue
-            
-            print(f"--- [Startup] ⏳ Downloading '{filename}' from repo '{repo_id}'... ---")
-            try:
-                cached_path = hf_hub_download(repo_id=repo_id, filename=repo_path)
-                os.makedirs(dest_dir, exist_ok=True)
-                os.symlink(cached_path, dest_path)
-                print(f"--- [Startup] ✅ Successfully downloaded and linked '{filename}'. ---")
-            except Exception as e:
-                print(f"--- [Startup] ❌ ERROR: Failed to download '{filename}': {e}")
-                print("--- [Startup] ❌ Please ensure your HF_TOKEN is set correctly and has access to the repository. ---")
-
-    finally:
-        if 'HF_TOKEN' in os.environ:
-            del os.environ['HF_TOKEN']
-            print("--- [Startup] ✅ Cleared HF_TOKEN environment variable. ---")
-            print(f"--- [Startup] Verifying HF_TOKEN after clearing: {os.environ.get('HF_TOKEN')}")
-        else:
-            print("--- [Startup] Note: HF_TOKEN environment variable was not set. Private downloads may fail without it. ---")
 
 def main():
     from utils.app_utils import print_welcome_message
     from scripts import build_sage_attention
+    from comfy_integration import setup as setup_comfyui
 
     print_welcome_message()
     
-    # Handle downloads that require authentication first.
-    handle_private_downloads()
-    
     print("--- [Setup] Attempting to build and install SageAttention... ---")
     try:
         build_sage_attention.install_sage_attention()
@@ -152,7 +59,9 @@ def main():
     except Exception as e:
         print(f"--- [Setup] ❌ SageAttention installation failed: {e}. Continuing with default attention. ---")
 
-
+    print("--- [Setup] Starting ComfyUI initialization ---")
+    setup_comfyui.initialize_comfyui()
+    
     print("--- [Setup] Reloading site-packages to detect newly installed packages... ---")
     try:
         site.main()
@@ -160,52 +69,20 @@ def main():
     except Exception as e:
         print(f"--- [Setup] ⚠️  Warning: Could not fully reload site-packages: {e} ---")
 
-    from comfy_integration import setup as setup_comfyui
-    from utils.app_utils import (
-        build_preprocessor_model_map,
-        build_preprocessor_parameter_map
-    )
-    from core import shared_state
-    from core.settings import ALL_MODEL_MAP, ALL_FILE_DOWNLOAD_MAP
-
-    def check_all_model_urls_on_startup():
-        print("--- [Setup] Checking all model URL validity (one-time check) ---")
-        for display_name, model_info in ALL_MODEL_MAP.items():
-            _, components, _, _ = model_info
-            if not components: continue
-            
-            for filename in components.values():
-                download_info = ALL_FILE_DOWNLOAD_MAP.get(filename, {})
-                repo_id = download_info.get('repo_id')
-                if not repo_id: continue
-                
-                repo_file_path = download_info.get('repository_file_path', filename)
-                url = f"https://huggingface.co/{repo_id}/resolve/main/{repo_file_path}"
-                
-                try:
-                    response = requests.head(url, timeout=5, allow_redirects=True)
-                    if response.status_code >= 400:
-                        print(f"❌ Invalid URL for '{display_name}' component '{filename}': {url} (Status: {response.status_code})")
-                        shared_state.INVALID_MODEL_URLS[display_name] = True
-                        break 
-                except requests.RequestException as e:
-                    print(f"❌ URL check failed for '{display_name}' component '{filename}': {e}")
-                    shared_state.INVALID_MODEL_URLS[display_name] = True
-                    break
-        print("--- [Setup] ✅ Finished checking model URLs. ---")
+    print("--- Initiating GPU Startup Check & SageAttention Patch ---")
+    try:
+        dummy_gpu_for_startup()
+    except Exception as e:
+        print(f"--- [GPU Startup] ⚠️ Warning: Startup check failed: {e} ---")
+
+    from utils.app_utils import load_ipadapter_presets
 
     print("--- Starting Application Setup ---")
     
-    setup_comfyui.initialize_comfyui()
+    print("--- Loading IPAdapter presets ---")
+    load_ipadapter_presets()
+    print("--- ✅ IPAdapter setup complete. ---")
 
-    check_all_model_urls_on_startup()
-    
-    print("--- Building ControlNet preprocessor maps ---")
-    from core.generation_logic import build_reverse_map
-    build_reverse_map()
-    build_preprocessor_model_map()
-    build_preprocessor_parameter_map()
-    print("--- ✅ ControlNet preprocessor setup complete. ---")
 
     print("--- Environment configured. Proceeding with module imports. ---")
     from ui.layout import build_ui

chain_injectors/controlnet_injector.py

@@ -0,0 +1,60 @@
+def inject(assembler, chain_definition, chain_items):
+    if not chain_items:
+        return
+
+    ksampler_name = chain_definition.get('ksampler_node', 'ksampler')
+    if ksampler_name not in assembler.node_map:
+        print(f"Warning: Target node '{ksampler_name}' for ControlNet chain not found. Skipping chain injection.")
+        return
+        
+    ksampler_id = assembler.node_map[ksampler_name]
+
+    if 'positive' not in assembler.workflow[ksampler_id]['inputs'] or \
+       'negative' not in assembler.workflow[ksampler_id]['inputs']:
+        print(f"Warning: KSampler node '{ksampler_name}' is missing 'positive' or 'negative' inputs. Skipping ControlNet chain.")
+        return
+
+    vae_source_str = chain_definition.get('vae_source')
+    if not vae_source_str:
+        print("Warning: 'vae_source' definition missing in the recipe for the ControlNet chain. Skipping.")
+        return
+    vae_node_name, vae_idx_str = vae_source_str.split(':')
+    if vae_node_name not in assembler.node_map:
+        print(f"Warning: VAE source node '{vae_node_name}' for ControlNet chain not found. Skipping.")
+        return
+    vae_connection = [assembler.node_map[vae_node_name], int(vae_idx_str)]
+        
+    current_positive_connection = assembler.workflow[ksampler_id]['inputs']['positive']
+    current_negative_connection = assembler.workflow[ksampler_id]['inputs']['negative']
+    
+    for item_data in chain_items:
+        cn_loader_id = assembler._get_unique_id()
+        cn_loader_node = assembler._get_node_template("ControlNetLoader")
+        cn_loader_node['inputs']['control_net_name'] = item_data['control_net_name']
+        assembler.workflow[cn_loader_id] = cn_loader_node
+
+        image_loader_id = assembler._get_unique_id()
+        image_loader_node = assembler._get_node_template("LoadImage")
+        image_loader_node['inputs']['image'] = item_data['image']
+        assembler.workflow[image_loader_id] = image_loader_node
+
+        apply_cn_id = assembler._get_unique_id()
+        apply_cn_node = assembler._get_node_template(chain_definition['template'])
+        
+        apply_cn_node['inputs']['strength'] = item_data['strength']
+        
+        apply_cn_node['inputs']['positive'] = current_positive_connection
+        apply_cn_node['inputs']['negative'] = current_negative_connection
+        apply_cn_node['inputs']['control_net'] = [cn_loader_id, 0]
+        apply_cn_node['inputs']['image'] = [image_loader_id, 0]
+        apply_cn_node['inputs']['vae'] = vae_connection
+        
+        assembler.workflow[apply_cn_id] = apply_cn_node
+
+        current_positive_connection = [apply_cn_id, 0]
+        current_negative_connection = [apply_cn_id, 1]
+
+    assembler.workflow[ksampler_id]['inputs']['positive'] = current_positive_connection
+    assembler.workflow[ksampler_id]['inputs']['negative'] = current_negative_connection
+    
+    print(f"ControlNet injector applied. KSampler inputs redirected through {len(chain_items)} ControlNet nodes.")

chain_injectors/diffsynth_controlnet_injector.py

@@ -0,0 +1,75 @@
+def inject(assembler, chain_definition, chain_items):
+    if not chain_items:
+        return
+
+    model_sampler_name = chain_definition.get('model_sampler_node')
+    ksampler_name = chain_definition.get('ksampler_node', 'ksampler')
+    
+    target_node_id = None
+    target_input_name = 'model'
+    
+    if model_sampler_name and model_sampler_name in assembler.node_map:
+        model_sampler_id = assembler.node_map[model_sampler_name]
+        if target_input_name in assembler.workflow[model_sampler_id]['inputs']:
+            target_node_id = model_sampler_id
+            print(f"ControlNet Model Patch injector targeting ModelSamplingAuraFlow node '{model_sampler_name}'.")
+
+    if not target_node_id:
+        if ksampler_name in assembler.node_map:
+            ksampler_id = assembler.node_map[ksampler_name]
+            if target_input_name in assembler.workflow[ksampler_id]['inputs']:
+                target_node_id = ksampler_id
+                print(f"ControlNet Model Patch injector targeting KSampler node '{ksampler_name}'.")
+        else:
+            print(f"Warning: Neither ModelSamplingAuraFlow node '{model_sampler_name}' nor KSampler node '{ksampler_name}' found for ControlNet patch chain. Skipping.")
+            return
+            
+    if not target_node_id:
+        print(f"Warning: Could not find a valid 'model' input on target nodes. Skipping ControlNet patch chain.")
+        return
+
+    current_model_connection = assembler.workflow[target_node_id]['inputs'][target_input_name]
+    
+    vae_source_str = chain_definition.get('vae_source')
+    vae_connection = None
+    if vae_source_str:
+        try:
+            vae_node_name, vae_idx_str = vae_source_str.split(':')
+            if vae_node_name in assembler.node_map:
+                vae_connection = [assembler.node_map[vae_node_name], int(vae_idx_str)]
+            else:
+                print(f"Warning: VAE source node '{vae_node_name}' not found for ControlNet patch chain. VAE will not be connected.")
+        except ValueError:
+            print(f"Warning: Invalid 'vae_source' format '{vae_source_str}' for ControlNet patch chain. Expected 'node_name:index'. VAE will not be connected.")
+    else:
+        print(f"Warning: 'vae_source' not defined for ControlNet patch chain definition. VAE may not be connected.")
+
+    for item_data in chain_items:
+        patch_loader_id = assembler._get_unique_id()
+        patch_loader_node = assembler._get_node_template("ModelPatchLoader")
+        patch_loader_node['inputs']['name'] = item_data['control_net_name']
+        assembler.workflow[patch_loader_id] = patch_loader_node
+
+        image_loader_id = assembler._get_unique_id()
+        image_loader_node = assembler._get_node_template("LoadImage")
+        image_loader_node['inputs']['image'] = item_data['image']
+        assembler.workflow[image_loader_id] = image_loader_node
+
+        apply_cn_id = assembler._get_unique_id()
+        apply_cn_node = assembler._get_node_template(chain_definition['template'])
+        
+        apply_cn_node['inputs']['strength'] = item_data.get('strength', 1.0)
+        apply_cn_node['inputs']['model'] = current_model_connection
+        apply_cn_node['inputs']['model_patch'] = [patch_loader_id, 0]
+        apply_cn_node['inputs']['image'] = [image_loader_id, 0]
+        
+        if 'vae' in apply_cn_node['inputs'] and vae_connection:
+            apply_cn_node['inputs']['vae'] = vae_connection
+        
+        assembler.workflow[apply_cn_id] = apply_cn_node
+
+        current_model_connection = [apply_cn_id, 0]
+
+    assembler.workflow[target_node_id]['inputs'][target_input_name] = current_model_connection
+    
+    print(f"ControlNet Model Patch injector applied. Target 'model' input re-routed through {len(chain_items)} patch(es).")

chain_injectors/flux1_ipadapter_injector.py

@@ -0,0 +1,46 @@
+def inject(assembler, chain_definition, chain_items):
+    if not chain_items:
+        return
+
+    ksampler_name = chain_definition.get('ksampler_node', 'ksampler')
+    if ksampler_name not in assembler.node_map:
+        print(f"Warning: KSampler node '{ksampler_name}' not found for Flux1 IPAdapter chain. Skipping.")
+        return
+        
+    ksampler_id = assembler.node_map[ksampler_name]
+
+    if 'model' not in assembler.workflow[ksampler_id]['inputs']:
+        print(f"Warning: KSampler node '{ksampler_name}' is missing 'model' input. Skipping Flux1 IPAdapter chain.")
+        return
+        
+    current_model_connection = assembler.workflow[ksampler_id]['inputs']['model']
+    
+    for item_data in chain_items:
+        image_loader_id = assembler._get_unique_id()
+        image_loader_node = assembler._get_node_template("LoadImage")
+        image_loader_node['inputs']['image'] = item_data['image']
+        assembler.workflow[image_loader_id] = image_loader_node
+
+        ipadapter_loader_id = assembler._get_unique_id()
+        ipadapter_loader_node = assembler._get_node_template("IPAdapterFluxLoader")
+        ipadapter_loader_node['inputs']['ipadapter'] = "ip-adapter.bin"
+        ipadapter_loader_node['inputs']['clip_vision'] = "google/siglip-so400m-patch14-384"
+        ipadapter_loader_node['inputs']['provider'] = "cuda"
+        assembler.workflow[ipadapter_loader_id] = ipadapter_loader_node
+        
+        apply_ipa_id = assembler._get_unique_id()
+        apply_ipa_node = assembler._get_node_template("ApplyIPAdapterFlux")
+        
+        apply_ipa_node['inputs']['weight'] = item_data['weight']
+        apply_ipa_node['inputs']['start_percent'] = item_data.get('start_percent', 0.0)
+        apply_ipa_node['inputs']['end_percent'] = item_data.get('end_percent', 0.6)
+        
+        apply_ipa_node['inputs']['model'] = current_model_connection
+        apply_ipa_node['inputs']['ipadapter_flux'] = [ipadapter_loader_id, 0]
+        apply_ipa_node['inputs']['image'] = [image_loader_id, 0]
+        
+        assembler.workflow[apply_ipa_id] = apply_ipa_node
+        current_model_connection = [apply_ipa_id, 0]
+
+    assembler.workflow[ksampler_id]['inputs']['model'] = current_model_connection
+    print(f"Flux1 IPAdapter injector applied. KSampler model input re-routed through {len(chain_items)} IPAdapter(s).")

chain_injectors/ipadapter_injector.py

@@ -0,0 +1,106 @@
+def inject(assembler, chain_definition, chain_items):
+    if not chain_items:
+        return
+
+    final_settings = {}
+    if chain_items and isinstance(chain_items[-1], dict) and chain_items[-1].get('is_final_settings'):
+        final_settings = chain_items.pop()
+
+    if not chain_items:
+        return
+
+    end_node_name = chain_definition.get('end')
+    if not end_node_name or end_node_name not in assembler.node_map:
+        print(f"Warning: Target node '{end_node_name}' for IPAdapter chain not found. Skipping chain injection.")
+        return
+        
+    end_node_id = assembler.node_map[end_node_name]
+    
+    if 'model' not in assembler.workflow[end_node_id]['inputs']:
+        print(f"Warning: Target node '{end_node_name}' is missing 'model' input. Skipping IPAdapter chain.")
+        return
+    
+    current_model_connection = assembler.workflow[end_node_id]['inputs']['model']
+    
+    model_type = final_settings.get('model_type', 'sdxl')
+    megapixels = 1.05 if model_type == 'sdxl' else 0.39
+
+    pos_embed_outputs = []
+    neg_embed_outputs = []
+
+    for i, item_data in enumerate(chain_items):
+        loader_type = 'FaceID' if 'FACEID' in item_data.get('preset', '') else 'Unified'
+        
+        loader_template_name = "IPAdapterUnifiedLoader"
+        if loader_type == 'FaceID':
+            loader_template_name = "IPAdapterUnifiedLoaderFaceID"
+
+        image_loader_id = assembler._get_unique_id()
+        image_loader_node = assembler._get_node_template("LoadImage")
+        image_loader_node['inputs']['image'] = item_data['image']
+        assembler.workflow[image_loader_id] = image_loader_node
+        
+        image_scaler_id = assembler._get_unique_id()
+        image_scaler_node = assembler._get_node_template("ImageScaleToTotalPixels")
+        image_scaler_node['inputs']['image'] = [image_loader_id, 0]
+        image_scaler_node['inputs']['megapixels'] = megapixels
+        image_scaler_node['inputs']['upscale_method'] = "lanczos"
+        assembler.workflow[image_scaler_id] = image_scaler_node
+
+        ipadapter_loader_id = assembler._get_unique_id()
+        ipadapter_loader_node = assembler._get_node_template(loader_template_name)
+        ipadapter_loader_node['inputs']['model'] = current_model_connection
+        ipadapter_loader_node['inputs']['preset'] = item_data['preset']
+        if loader_type == 'FaceID':
+             ipadapter_loader_node['inputs']['lora_strength'] = item_data.get('lora_strength', 0.6)
+        assembler.workflow[ipadapter_loader_id] = ipadapter_loader_node
+        
+        encoder_id = assembler._get_unique_id()
+        encoder_node = assembler._get_node_template("IPAdapterEncoder")
+        encoder_node['inputs']['weight'] = item_data['weight']
+        encoder_node['inputs']['ipadapter'] = [ipadapter_loader_id, 1]
+        encoder_node['inputs']['image'] = [image_scaler_id, 0]
+        assembler.workflow[encoder_id] = encoder_node
+        
+        pos_embed_outputs.append([encoder_id, 0])
+        neg_embed_outputs.append([encoder_id, 1])
+
+    pos_combiner_id = assembler._get_unique_id()
+    pos_combiner_node = assembler._get_node_template("IPAdapterCombineEmbeds")
+    pos_combiner_node['inputs']['method'] = final_settings.get('final_combine_method', 'concat')
+    for i, conn in enumerate(pos_embed_outputs):
+        pos_combiner_node['inputs'][f'embed{i+1}'] = conn
+    assembler.workflow[pos_combiner_id] = pos_combiner_node
+
+    neg_combiner_id = assembler._get_unique_id()
+    neg_combiner_node = assembler._get_node_template("IPAdapterCombineEmbeds")
+    neg_combiner_node['inputs']['method'] = final_settings.get('final_combine_method', 'concat')
+    for i, conn in enumerate(neg_embed_outputs):
+        neg_combiner_node['inputs'][f'embed{i+1}'] = conn
+    assembler.workflow[neg_combiner_id] = neg_combiner_node
+    
+    final_loader_type = 'FaceID' if 'FACEID' in final_settings.get('final_preset', '') else 'Unified'
+    final_loader_template_name = "IPAdapterUnifiedLoader"
+    if final_loader_type == 'FaceID':
+        final_loader_template_name = "IPAdapterUnifiedLoaderFaceID"
+
+    final_loader_id = assembler._get_unique_id()
+    final_loader_node = assembler._get_node_template(final_loader_template_name)
+    final_loader_node['inputs']['model'] = current_model_connection
+    final_loader_node['inputs']['preset'] = final_settings.get('final_preset', 'STANDARD (medium strength)')
+    if final_loader_type == 'FaceID':
+        final_loader_node['inputs']['lora_strength'] = final_settings.get('final_lora_strength', 0.6)
+    assembler.workflow[final_loader_id] = final_loader_node
+
+    apply_embeds_id = assembler._get_unique_id()
+    apply_embeds_node = assembler._get_node_template("IPAdapterEmbeds")
+    apply_embeds_node['inputs']['weight'] = final_settings.get('final_weight', 1.0)
+    apply_embeds_node['inputs']['embeds_scaling'] = final_settings.get('final_embeds_scaling', 'V only')
+    apply_embeds_node['inputs']['model'] = [final_loader_id, 0]
+    apply_embeds_node['inputs']['ipadapter'] = [final_loader_id, 1]
+    apply_embeds_node['inputs']['pos_embed'] = [pos_combiner_id, 0]
+    apply_embeds_node['inputs']['neg_embed'] = [neg_combiner_id, 0]
+    assembler.workflow[apply_embeds_id] = apply_embeds_node
+
+    assembler.workflow[end_node_id]['inputs']['model'] = [apply_embeds_id, 0]
+    print(f"IPAdapter injector applied. Redirected '{end_node_name}' model input through {len(chain_items)} reference images.")

chain_injectors/newbie_lora_injector.py

@@ -0,0 +1,63 @@
+from copy import deepcopy
+
+def inject(assembler, chain_definition, chain_items):
+    if not chain_items:
+        return
+
+    output_map = chain_definition.get('output_map', {})
+    current_connections = {}
+    for key, type_name in output_map.items():
+        if ':' in str(key):
+            node_name, idx_str = key.split(':')
+            if node_name not in assembler.node_map:
+                print(f"Warning: [NewBie LoRA Injector] Node '{node_name}' in chain's output_map not found. Skipping.")
+                continue
+            node_id = assembler.node_map[node_name]
+            start_output_idx = int(idx_str)
+            current_connections[type_name] = [node_id, start_output_idx]
+        else:
+            print(f"Warning: [NewBie LoRA Injector] output_map key '{key}' is not in 'node:index' format. Skipping this connection.")
+
+    template_name = chain_definition.get('template')
+    if not template_name:
+        print(f"Warning: [NewBie LoRA Injector] No 'template' defined for chain. Skipping.")
+        return
+
+    for item_data in chain_items:
+        template = assembler._get_node_template(template_name)
+        node_data = deepcopy(template)
+        
+        node_data['inputs']['lora_name'] = item_data.get('lora_name')
+        node_data['inputs']['strength'] = item_data.get('strength_model', 1.0)
+        node_data['inputs']['enabled'] = True
+        
+        if 'model' in current_connections:
+            node_data['inputs']['model'] = current_connections['model']
+        if 'clip' in current_connections:
+            node_data['inputs']['clip'] = current_connections['clip']
+
+        new_node_id = assembler._get_unique_id()
+        assembler.workflow[new_node_id] = node_data
+        
+        current_connections['model'] = [new_node_id, 0]
+        current_connections['clip'] = [new_node_id, 1]
+
+    end_input_map = chain_definition.get('end_input_map', {})
+    for type_name, targets in end_input_map.items():
+        if type_name in current_connections:
+            if not isinstance(targets, list):
+                targets = [targets]
+            
+            for target_str in targets:
+                try:
+                    end_node_name, end_input_name = target_str.split(':')
+                    if end_node_name in assembler.node_map:
+                        end_node_id = assembler.node_map[end_node_name]
+                        assembler.workflow[end_node_id]['inputs'][end_input_name] = current_connections[type_name]
+                    else:
+                        print(f"Warning: [NewBie LoRA Injector] End node '{end_node_name}' for dynamic chain not found. Skipping connection.")
+                except ValueError:
+                    print(f"Warning: [NewBie LoRA Injector] Invalid target format '{target_str}' in end_input_map. Skipping.")
+
+    if chain_items:
+        print(f"NewBie LoRA injector applied. Re-routed model and clip through {len(chain_items)} LoRA(s).")

chain_injectors/reference_latent_injector.py

@@ -2,15 +2,78 @@ def inject(assembler, chain_definition, chain_items):
     if not chain_items:
         return
 
-    ksampler_name = chain_definition.get('ksampler_node', 'ksampler')
-    flux_guidance_name = chain_definition.get('flux_guidance_node')
+    guider_node_name = chain_definition.get('guider_node')
+    guider_target_inputs = chain_definition.get('guider_target_inputs', [])
+    start_connections_map = chain_definition.get('start_connections', {})
     vae_node_name = chain_definition.get('vae_node', 'vae_loader')
 
+    if guider_node_name and guider_node_name in assembler.node_map and guider_target_inputs:
+        guider_id = assembler.node_map[guider_node_name]
+        if vae_node_name not in assembler.node_map:
+            print(f"Warning: VAE node '{vae_node_name}' not found for Guider chain. Skipping.")
+            return
+        vae_node_id = assembler.node_map[vae_node_name]
+        
+        print(f"ReferenceLatent injector targeting DualCFGGuider node '{guider_node_name}'.")
+
+        current_connections = {}
+        for target_input in guider_target_inputs:
+            conn_str = start_connections_map.get(target_input)
+            if not conn_str:
+                print(f"Warning: No start connection defined for '{target_input}' in Guider chain. Skipping this input.")
+                continue
+            try:
+                node_name, idx_str = conn_str.split(':')
+                node_id = assembler.node_map[node_name]
+                current_connections[target_input] = [node_id, int(idx_str)]
+            except (ValueError, KeyError):
+                print(f"Warning: Invalid start connection '{conn_str}' for '{target_input}'. Skipping.")
+
+        encoded_latents = []
+        for i, img_filename in enumerate(chain_items):
+            load_id = assembler._get_unique_id()
+            load_node = assembler._get_node_template("LoadImage")
+            load_node['inputs']['image'] = img_filename
+            assembler.workflow[load_id] = load_node
+            
+            scale_id = assembler._get_unique_id()
+            scale_node = assembler._get_node_template("ImageScaleToTotalPixels")
+            scale_node['inputs']['megapixels'] = 1.0
+            scale_node['inputs']['upscale_method'] = "lanczos"
+            scale_node['inputs']['image'] = [load_id, 0]
+            assembler.workflow[scale_id] = scale_node
+            
+            vae_encode_id = assembler._get_unique_id()
+            vae_encode_node = assembler._get_node_template("VAEEncode")
+            vae_encode_node['inputs']['pixels'] = [scale_id, 0]
+            vae_encode_node['inputs']['vae'] = [vae_node_id, 0]
+            assembler.workflow[vae_encode_id] = vae_encode_node
+            encoded_latents.append([vae_encode_id, 0])
+        
+        for target_input_name, start_connection in current_connections.items():
+            current_chain_head = start_connection
+            for i, latent_conn in enumerate(encoded_latents):
+                ref_latent_id = assembler._get_unique_id()
+                ref_latent_node = assembler._get_node_template("ReferenceLatent")
+                ref_latent_node['inputs']['conditioning'] = current_chain_head
+                ref_latent_node['inputs']['latent'] = latent_conn
+                ref_latent_node['_meta']['title'] = f"{target_input_name} RefLatent {i+1}"
+                assembler.workflow[ref_latent_id] = ref_latent_node
+                current_chain_head = [ref_latent_id, 0]
+            
+            assembler.workflow[guider_id]['inputs'][target_input_name] = current_chain_head
+            print(f"  - Input '{target_input_name}' of node '{guider_node_name}' re-routed through {len(chain_items)} reference images.")
+
+        return
+
+    flux_guidance_name = chain_definition.get('flux_guidance_node')
+    ksampler_name = chain_definition.get('ksampler_node', 'ksampler')
+
     if ksampler_name not in assembler.node_map:
-        print(f"Warning: [ReferenceLatent] KSampler node '{ksampler_name}' not found. Skipping.")
+        print(f"Warning: KSampler node '{ksampler_name}' not found for ReferenceLatent chain. Skipping.")
         return
     if vae_node_name not in assembler.node_map:
-        print(f"Warning: [ReferenceLatent] VAE loader node '{vae_node_name}' not found. Skipping.")
+        print(f"Warning: VAE loader node '{vae_node_name}' not found for ReferenceLatent chain. Skipping.")
         return
 
     ksampler_id = assembler.node_map[ksampler_name]
@@ -23,44 +86,72 @@ def inject(assembler, chain_definition, chain_items):
         if 'conditioning' in assembler.workflow[flux_guidance_id]['inputs']:
             pos_target_node_id = flux_guidance_id
             pos_target_input_name = 'conditioning'
-            print(f"ReferenceLatent injector targeting FluxGuidance node '{flux_guidance_name}'.")
+            print(f"ReferenceLatent injector targeting FluxGuidance node '{flux_guidance_name}' for positive chain.")
     
     if not pos_target_node_id:
         if 'positive' in assembler.workflow[ksampler_id]['inputs']:
             pos_target_node_id = ksampler_id
             pos_target_input_name = 'positive'
-            print(f"ReferenceLatent injector targeting KSampler node '{ksampler_name}'.")
+            print(f"ReferenceLatent injector targeting KSampler node '{ksampler_name}' for positive chain.")
         else:
-            print(f"Warning: [ReferenceLatent] Could not find a valid positive injection point. Skipping.")
+            print(f"Warning: Could not find a valid positive injection point for ReferenceLatent chain. Skipping.")
             return
 
     current_pos_conditioning = assembler.workflow[pos_target_node_id]['inputs'][pos_target_input_name]
 
-    for i, img_filename in enumerate(chain_items):
-        if not img_filename or not isinstance(img_filename, str):
-            continue
+    neg_target_node_id = ksampler_id
+    neg_target_input_name = 'negative'
+    if 'negative' not in assembler.workflow[neg_target_node_id]['inputs']:
+        print(f"Warning: KSampler node '{ksampler_name}' has no 'negative' input. Skipping negative ReferenceLatent chain.")
+        neg_target_node_id = None
+    
+    current_neg_conditioning = None
+    if neg_target_node_id:
+        current_neg_conditioning = assembler.workflow[neg_target_node_id]['inputs'][neg_target_input_name]
 
+    for i, img_filename in enumerate(chain_items):
         load_id = assembler._get_unique_id()
         load_node = assembler._get_node_template("LoadImage")
         load_node['inputs']['image'] = img_filename
+        load_node['_meta']['title'] = f"Load Reference Image {i+1}"
         assembler.workflow[load_id] = load_node
         
+        scale_id = assembler._get_unique_id()
+        scale_node = assembler._get_node_template("ImageScaleToTotalPixels")
+        scale_node['inputs']['megapixels'] = 1.0
+        scale_node['inputs']['upscale_method'] = "lanczos"
+        scale_node['inputs']['image'] = [load_id, 0]
+        scale_node['_meta']['title'] = f"Scale Reference {i+1}"
+        assembler.workflow[scale_id] = scale_node
+        
         vae_encode_id = assembler._get_unique_id()
         vae_encode_node = assembler._get_node_template("VAEEncode")
-        vae_encode_node['inputs']['pixels'] = [load_id, 0]
+        vae_encode_node['inputs']['pixels'] = [scale_id, 0]
         vae_encode_node['inputs']['vae'] = [vae_node_id, 0]
+        vae_encode_node['_meta']['title'] = f"VAE Encode Reference {i+1}"
         assembler.workflow[vae_encode_id] = vae_encode_node
         
         latent_conn = [vae_encode_id, 0]
 
-        ref_latent_id = assembler._get_unique_id()
-        ref_latent_node = assembler._get_node_template("ReferenceLatent")
-        ref_latent_node['inputs']['conditioning'] = current_pos_conditioning
-        ref_latent_node['inputs']['latent'] = latent_conn
-        assembler.workflow[ref_latent_id] = ref_latent_node
-        
-        current_pos_conditioning = [ref_latent_id, 0]
+        pos_ref_latent_id = assembler._get_unique_id()
+        pos_ref_latent_node = assembler._get_node_template("ReferenceLatent")
+        pos_ref_latent_node['inputs']['conditioning'] = current_pos_conditioning
+        pos_ref_latent_node['inputs']['latent'] = latent_conn
+        pos_ref_latent_node['_meta']['title'] = f"Positive ReferenceLatent {i+1}"
+        assembler.workflow[pos_ref_latent_id] = pos_ref_latent_node
+        current_pos_conditioning = [pos_ref_latent_id, 0]
+
+        if neg_target_node_id:
+            neg_ref_latent_id = assembler._get_unique_id()
+            neg_ref_latent_node = assembler._get_node_template("ReferenceLatent")
+            neg_ref_latent_node['inputs']['conditioning'] = current_neg_conditioning
+            neg_ref_latent_node['inputs']['latent'] = latent_conn
+            neg_ref_latent_node['_meta']['title'] = f"Negative ReferenceLatent {i+1}"
+            assembler.workflow[neg_ref_latent_id] = neg_ref_latent_node
+            current_neg_conditioning = [neg_ref_latent_id, 0]
 
     assembler.workflow[pos_target_node_id]['inputs'][pos_target_input_name] = current_pos_conditioning
+    if neg_target_node_id:
+        assembler.workflow[neg_target_node_id]['inputs'][neg_target_input_name] = current_neg_conditioning
     
-    print(f"ReferenceLatent injector applied. Re-routed inputs through {len(chain_items)} reference image(s).")
+    print(f"ReferenceLatent injector applied. Re-routed inputs through {len(chain_items)} reference images.")

chain_injectors/sd3_ipadapter_injector.py

@@ -0,0 +1,66 @@
+def inject(assembler, chain_definition, chain_items):
+    if not chain_items:
+        return
+
+    ksampler_name = chain_definition.get('ksampler_node', 'ksampler')
+    if ksampler_name not in assembler.node_map:
+        print(f"Warning: KSampler node '{ksampler_name}' not found for SD3 IPAdapter chain. Skipping.")
+        return
+        
+    ksampler_id = assembler.node_map[ksampler_name]
+
+    if 'model' not in assembler.workflow[ksampler_id]['inputs']:
+        print(f"Warning: KSampler node '{ksampler_name}' is missing 'model' input. Skipping SD3 IPAdapter chain.")
+        return
+        
+    current_model_connection = assembler.workflow[ksampler_id]['inputs']['model']
+    
+    clip_vision_loader_id = assembler._get_unique_id()
+    clip_vision_loader_node = assembler._get_node_template("CLIPVisionLoader")
+    clip_vision_loader_node['inputs']['clip_name'] = "sigclip_vision_patch14_384.safetensors"
+    assembler.workflow[clip_vision_loader_id] = clip_vision_loader_node
+
+    ipadapter_loader_id = assembler._get_unique_id()
+    ipadapter_loader_node = assembler._get_node_template("IPAdapterSD3Loader")
+    ipadapter_loader_node['inputs']['ipadapter'] = "ip-adapter_sd35l_instantx.bin"
+    ipadapter_loader_node['inputs']['provider'] = "cuda"
+    assembler.workflow[ipadapter_loader_id] = ipadapter_loader_node
+
+    for item_data in chain_items:
+        image_loader_id = assembler._get_unique_id()
+        image_loader_node = assembler._get_node_template("LoadImage")
+        image_loader_node['inputs']['image'] = item_data['image']
+        assembler.workflow[image_loader_id] = image_loader_node
+
+        image_scaler_id = assembler._get_unique_id()
+        image_scaler_node = assembler._get_node_template("ImageScaleToTotalPixels")
+        image_scaler_node['inputs']['image'] = [image_loader_id, 0]
+        image_scaler_node['inputs']['upscale_method'] = 'nearest-exact'
+        image_scaler_node['inputs']['megapixels'] = 1.0
+        assembler.workflow[image_scaler_id] = image_scaler_node
+
+        clip_vision_encode_id = assembler._get_unique_id()
+        clip_vision_encode_node = assembler._get_node_template("CLIPVisionEncode")
+        clip_vision_encode_node['inputs']['crop'] = "center"
+        clip_vision_encode_node['inputs']['clip_vision'] = [clip_vision_loader_id, 0]
+        clip_vision_encode_node['inputs']['image'] = [image_scaler_id, 0]
+        assembler.workflow[clip_vision_encode_id] = clip_vision_encode_node
+        
+        apply_ipa_id = assembler._get_unique_id()
+        apply_ipa_node = assembler._get_node_template("ApplyIPAdapterSD3")
+        
+        apply_ipa_node['inputs']['weight'] = item_data.get('weight', 1.0)
+        apply_ipa_node['inputs']['start_percent'] = item_data.get('start_percent', 0.0)
+        apply_ipa_node['inputs']['end_percent'] = item_data.get('end_percent', 1.0)
+        
+        apply_ipa_node['inputs']['model'] = current_model_connection
+        apply_ipa_node['inputs']['ipadapter'] = [ipadapter_loader_id, 0]
+        apply_ipa_node['inputs']['image_embed'] = [clip_vision_encode_id, 0]
+        
+        assembler.workflow[apply_ipa_id] = apply_ipa_node
+        
+        current_model_connection = [apply_ipa_id, 0]
+
+    assembler.workflow[ksampler_id]['inputs']['model'] = current_model_connection
+    
+    print(f"SD3 IPAdapter injector applied. KSampler model input re-routed through {len(chain_items)} IPAdapter(s).")

chain_injectors/style_injector.py

@@ -0,0 +1,71 @@
+def inject(assembler, chain_definition, chain_items):
+    if not chain_items:
+        return
+
+    flux_guidance_name = chain_definition.get('flux_guidance_node')
+    ksampler_name = chain_definition.get('ksampler_node', 'ksampler')
+
+    target_node_id = None
+    target_input_name = None
+    
+    if flux_guidance_name and flux_guidance_name in assembler.node_map:
+        flux_guidance_id = assembler.node_map[flux_guidance_name]
+        if 'conditioning' in assembler.workflow[flux_guidance_id]['inputs']:
+            target_node_id = flux_guidance_id
+            target_input_name = 'conditioning'
+
+    if not target_node_id:
+        if ksampler_name in assembler.node_map:
+            ksampler_id = assembler.node_map[ksampler_name]
+            if 'positive' in assembler.workflow[ksampler_id]['inputs']:
+                target_node_id = ksampler_id
+                target_input_name = 'positive'
+        else:
+            return
+            
+    if not target_node_id:
+        return
+
+    current_conditioning = assembler.workflow[target_node_id]['inputs'][target_input_name]
+
+    style_model_loader_id = assembler._get_unique_id()
+    style_model_loader_node = assembler._get_node_template("StyleModelLoader")
+    style_model_loader_node['inputs']['style_model_name'] = "flux1-redux-dev.safetensors"
+    assembler.workflow[style_model_loader_id] = style_model_loader_node
+
+    clip_vision_loader_id = assembler._get_unique_id()
+    clip_vision_loader_node = assembler._get_node_template("CLIPVisionLoader")
+    clip_vision_loader_node['inputs']['clip_name'] = "sigclip_vision_patch14_384.safetensors"
+    assembler.workflow[clip_vision_loader_id] = clip_vision_loader_node
+
+    for item_data in chain_items:
+        image = item_data.get('image')
+        strength = item_data.get('strength', 1.0)
+        if not image or strength is None:
+            continue
+
+        load_image_id = assembler._get_unique_id()
+        clip_vision_encode_id = assembler._get_unique_id()
+        style_apply_id = assembler._get_unique_id()
+
+        load_image_node = assembler._get_node_template("LoadImage")
+        clip_vision_encode_node = assembler._get_node_template("CLIPVisionEncode")
+        style_apply_node = assembler._get_node_template("StyleModelApply")
+        
+        load_image_node['inputs']['image'] = image
+        clip_vision_encode_node['inputs']['crop'] = "center"
+        clip_vision_encode_node['inputs']['clip_vision'] = [clip_vision_loader_id, 0]
+        clip_vision_encode_node['inputs']['image'] = [load_image_id, 0]
+
+        style_apply_node['inputs']['strength'] = strength
+        style_apply_node['inputs']['strength_type'] = "multiply"
+        style_apply_node['inputs']['conditioning'] = current_conditioning
+        style_apply_node['inputs']['style_model'] = [style_model_loader_id, 0]
+        style_apply_node['inputs']['clip_vision_output'] = [clip_vision_encode_id, 0]
+        
+        assembler.workflow[load_image_id] = load_image_node
+        assembler.workflow[clip_vision_encode_id] = clip_vision_encode_node
+        assembler.workflow[style_apply_id] = style_apply_node
+        current_conditioning = [style_apply_id, 0]
+
+    assembler.workflow[target_node_id]['inputs'][target_input_name] = current_conditioning

chain_injectors/vae_injector.py

@@ -0,0 +1,30 @@
+def inject(assembler, chain_definition, chain_items):
+    if not chain_items:
+        return
+
+    vae_name = chain_items[0] if isinstance(chain_items, list) else chain_items
+    if not vae_name or vae_name == "None":
+        return
+
+    targets = chain_definition.get('targets', [])
+    if not targets:
+        return
+
+    vae_loader_id = assembler._get_unique_id()
+    vae_loader_node = assembler._get_node_template("VAELoader")
+    vae_loader_node['inputs']['vae_name'] = vae_name
+    assembler.workflow[vae_loader_id] = vae_loader_node
+
+    injected_count = 0
+    for target_str in targets:
+        try:
+            node_name, input_name = target_str.split(':')
+            if node_name in assembler.node_map:
+                node_id = assembler.node_map[node_name]
+                assembler.workflow[node_id]['inputs'][input_name] = [vae_loader_id, 0]
+                injected_count += 1
+        except ValueError:
+            print(f"Warning: Invalid VAE injector target format '{target_str}'. Expected 'node_name:input_name'.")
+    
+    if injected_count > 0:
+        print(f"VAE injector applied. Rerouted {injected_count} connection(s) to new VAELoader ({vae_name}).")

comfy_integration/nodes.py

@@ -23,6 +23,11 @@ CLIPTextEncodeSDXL = NODE_CLASS_MAPPINGS['CLIPTextEncodeSDXL']
 LoraLoader = NODE_CLASS_MAPPINGS['LoraLoader']
 CLIPSetLastLayer = NODE_CLASS_MAPPINGS['CLIPSetLastLayer']
 
+if 'EmptyHunyuanImageLatent' in NODE_CLASS_MAPPINGS:
+    EmptyHunyuanImageLatent = NODE_CLASS_MAPPINGS['EmptyHunyuanImageLatent']
+else:
+    print("⚠️ Warning: 'EmptyHunyuanImageLatent' not found in NODE_CLASS_MAPPINGS. HunyuanImage txt2img may fail if this node is required.")
+
 try:
     KSamplerNode = NODE_CLASS_MAPPINGS['KSampler']
     SAMPLER_CHOICES = KSamplerNode.INPUT_TYPES()["required"]["sampler_name"][0]

comfy_integration/setup.py

@@ -39,14 +39,40 @@ def initialize_comfyui():
     except OSError as e:
         print(f"⚠️ Could not remove temporary directory '{COMFYUI_TEMP_DIR}': {e}")
 
+
     print("--- Cloning third-party extensions for ComfyUI ---")
-    controlnet_aux_path = os.path.join(APP_DIR, "custom_nodes", "comfyui_controlnet_aux")
-    if not os.path.exists(controlnet_aux_path):
-        os.system(f"git clone https://github.com/Fannovel16/comfyui_controlnet_aux.git {controlnet_aux_path}")
-        print("✅ comfyui_controlnet_aux extension cloned.")
+
+    # 1. ComfyUI_IPAdapter_plus
+    ipadapter_plus_path = os.path.join(APP_DIR, "custom_nodes", "ComfyUI_IPAdapter_plus")
+    if not os.path.exists(ipadapter_plus_path):
+        os.system(f"git clone https://github.com/cubiq/ComfyUI_IPAdapter_plus.git {ipadapter_plus_path}")
+        print("✅ ComfyUI_IPAdapter_plus extension cloned.")
     else:
-        print("✅ comfyui_controlnet_aux extension already exists.")
+        print("✅ ComfyUI_IPAdapter_plus extension already exists.")
 
+    # 2. ComfyUI-InstantX-IPAdapter-SD3
+    ipadapter_plus_path = os.path.join(APP_DIR, "custom_nodes", "ComfyUI-InstantX-IPAdapter-SD3")
+    if not os.path.exists(ipadapter_plus_path):
+        os.system(f"git clone https://github.com/Slickytail/ComfyUI-InstantX-IPAdapter-SD3.git {ipadapter_plus_path}")
+        print("✅ ComfyUI-InstantX-IPAdapter-SD3 extension cloned.")
+    else:
+        print("✅ ComfyUI-InstantX-IPAdapter-SD3 extension already exists.")
+
+    # 3. ComfyUI-IPAdapter-Flux
+    ipadapter_flux_path = os.path.join(APP_DIR, "custom_nodes", "ComfyUI-IPAdapter-Flux")
+    if not os.path.exists(ipadapter_flux_path):
+        os.system(f"git clone https://github.com/Shakker-Labs/ComfyUI-IPAdapter-Flux.git {ipadapter_flux_path}")
+        print("✅ ComfyUI-IPAdapter-Flux extension cloned.")
+    else:
+        print("✅ ComfyUI-IPAdapter-Flux extension already exists.")
+
+    # 4. ComfyUI-Newbie-Nodes
+    newbie_nodes_path = os.path.join(APP_DIR, "custom_nodes", "ComfyUI-Newbie-Nodes")
+    if not os.path.exists(newbie_nodes_path):
+        os.system(f"git clone https://github.com/NewBieAI-Lab/ComfyUI-Newbie-Nodes.git {newbie_nodes_path}")
+        print("✅ ComfyUI-Newbie-Nodes extension cloned.")
+    else:
+        print("✅ ComfyUI-Newbie-Nodes extension already exists.")
 
     print(f"✅ Current working directory is: {os.getcwd()}")
     
@@ -55,13 +81,10 @@ def initialize_comfyui():
 
     print("✅ ComfyUI initialized with default attention mechanism.")
 
-    os.makedirs(os.path.join(APP_DIR, CHECKPOINT_DIR), exist_ok=True)
-    os.makedirs(os.path.join(APP_DIR, LORA_DIR), exist_ok=True)
-    os.makedirs(os.path.join(APP_DIR, EMBEDDING_DIR), exist_ok=True)
-    os.makedirs(os.path.join(APP_DIR, CONTROLNET_DIR), exist_ok=True)
-    os.makedirs(os.path.join(APP_DIR, MODEL_PATCHES_DIR), exist_ok=True)
-    os.makedirs(os.path.join(APP_DIR, DIFFUSION_MODELS_DIR), exist_ok=True)
-    os.makedirs(os.path.join(APP_DIR, VAE_DIR), exist_ok=True)
-    os.makedirs(os.path.join(APP_DIR, TEXT_ENCODERS_DIR), exist_ok=True)
+    for dir_path in CATEGORY_TO_DIR_MAP.values():
+        os.makedirs(os.path.join(APP_DIR, dir_path), exist_ok=True)
+    
     os.makedirs(os.path.join(APP_DIR, INPUT_DIR), exist_ok=True)
+    os.makedirs(os.path.join(APP_DIR, OUTPUT_DIR), exist_ok=True)
+    
     print("✅ All required model directories are present.")

core/generation_logic.py

@@ -1,25 +1,10 @@
 from typing import Any, Dict
 import gradio as gr
 
-from core.pipelines.controlnet_preprocessor import ControlNetPreprocessorPipeline
 from core.pipelines.sd_image_pipeline import SdImagePipeline
 
-controlnet_preprocessor_pipeline = ControlNetPreprocessorPipeline()
 sd_image_pipeline = SdImagePipeline()
 
 
-def build_reverse_map():
-    from nodes import NODE_DISPLAY_NAME_MAPPINGS
-    import core.pipelines.controlnet_preprocessor as cn_module
-    
-    if cn_module.REVERSE_DISPLAY_NAME_MAP is None:
-        cn_module.REVERSE_DISPLAY_NAME_MAP = {v: k for k, v in NODE_DISPLAY_NAME_MAPPINGS.items()}
-        if "Semantic Segmentor (legacy, alias for UniFormer)" not in cn_module.REVERSE_DISPLAY_NAME_MAP:
-             cn_module.REVERSE_DISPLAY_NAME_MAP["Semantic Segmentor (legacy, alias for UniFormer)"] = "SemSegPreprocessor"
-
-
-def run_cn_preprocessor_entry(*args, **kwargs):
-    return controlnet_preprocessor_pipeline.run(*args, **kwargs)
-
 def generate_image_wrapper(ui_inputs: dict, progress=gr.Progress(track_tqdm=True)):
     return sd_image_pipeline.run(ui_inputs=ui_inputs, progress=progress)

core/model_manager.py

@@ -1,9 +1,8 @@
 import gc
 from typing import List
 import gradio as gr
-
-from core.settings import ALL_MODEL_MAP
 from utils.app_utils import _ensure_model_downloaded
+from core.settings import ALL_MODEL_MAP
 
 class ModelManager:
     _instance = None
@@ -21,25 +20,13 @@ class ModelManager:
 
     def ensure_models_downloaded(self, required_models: List[str], progress):
         print(f"--- [ModelManager] Ensuring models are downloaded: {required_models} ---")
-        
-        files_to_download = set()
-        for display_name in required_models:
-            if display_name in ALL_MODEL_MAP:
-                _, components, _, _ = ALL_MODEL_MAP[display_name]
-                for component_file in components.values():
-                    files_to_download.add(component_file)
-        
-        files_to_download = list(files_to_download)
-        total_files = len(files_to_download)
-        
-        for i, filename in enumerate(files_to_download):
+        for i, display_name in enumerate(required_models):
             if progress and hasattr(progress, '__call__'):
-                progress(i / total_files if total_files > 0 else 0, desc=f"Checking file: {filename}")
+                progress(i / max(len(required_models), 1), desc=f"Checking file: {display_name}")
             try:
-                _ensure_model_downloaded(filename, progress)
+                _ensure_model_downloaded(display_name, progress)
             except Exception as e:
-                raise gr.Error(f"Failed to download model component '{filename}'. Reason: {e}")
-
+                raise gr.Error(f"Failed to download model '{display_name}'. Reason: {e}")
         print(f"--- [ModelManager] ✅ All required models are present on disk. ---")
-
+    
 model_manager = ModelManager()

core/pipelines/controlnet_preprocessor.py

@@ -1,143 +0,0 @@
-from typing import Dict, Any, List
-import imageio
-import tempfile
-import numpy as np
-import torch
-import gradio as gr
-from PIL import Image
-import spaces
-
-from .base_pipeline import BasePipeline
-from comfy_integration.nodes import NODE_CLASS_MAPPINGS
-from nodes import NODE_DISPLAY_NAME_MAPPINGS
-from utils.app_utils import get_value_at_index
-
-REVERSE_DISPLAY_NAME_MAP = None
-CPU_ONLY_PREPROCESSORS = {
-    "Binary Lines", "Canny Edge", "Color Pallete", "Fake Scribble Lines (aka scribble_hed)",
-    "Image Intensity", "Image Luminance", "Inpaint Preprocessor", "PyraCanny", "Scribble Lines",
-    "Scribble XDoG Lines", "Standard Lineart", "Content Shuffle", "Tile"
-}
-
-def run_node_by_function_name(node_instance: Any, **kwargs) -> Any:
-    node_class = type(node_instance)
-    function_name = getattr(node_class, 'FUNCTION', None)
-    if not function_name:
-        raise AttributeError(f"Node class '{node_class.__name__}' is missing the required 'FUNCTION' attribute.")
-    execution_method = getattr(node_instance, function_name, None)
-    if not callable(execution_method):
-        raise AttributeError(f"Method '{function_name}' not found or not callable on node '{node_class.__name__}'.")
-    return execution_method(**kwargs)
-
-class ControlNetPreprocessorPipeline(BasePipeline):
-    def get_required_models(self, **kwargs) -> List[str]:
-        return []
-
-    def _gpu_logic(
-        self, pil_images: List[Image.Image], preprocessor_name: str, model_name: str,
-        params: Dict[str, Any], progress=gr.Progress(track_tqdm=True)
-    ) -> List[Image.Image]:
-        global REVERSE_DISPLAY_NAME_MAP
-        if REVERSE_DISPLAY_NAME_MAP is None:
-            raise RuntimeError("REVERSE_DISPLAY_NAME_MAP has not been initialized. `build_reverse_map` must be called on startup.")
-            
-        class_name = REVERSE_DISPLAY_NAME_MAP.get(preprocessor_name)
-        if not class_name or class_name not in NODE_CLASS_MAPPINGS:
-            raise ValueError(f"Preprocessor '{preprocessor_name}' not found.")
-            
-        preprocessor_instance = NODE_CLASS_MAPPINGS[class_name]()
-        call_args = {**params, 'ckpt_name': model_name}
-        
-        processed_pil_images = []
-        total_frames = len(pil_images)
-
-        for i, frame_pil in enumerate(pil_images):
-            progress(i / total_frames, desc=f"Processing frame {i+1}/{total_frames} with {preprocessor_name}...")
-            
-            frame_tensor = torch.from_numpy(np.array(frame_pil).astype(np.float32) / 255.0).unsqueeze(0)
-            
-            resolution_arg = {'resolution': max(frame_tensor.shape[2], frame_tensor.shape[3])}
-
-            result_tuple = run_node_by_function_name(
-                preprocessor_instance, 
-                image=frame_tensor, 
-                **resolution_arg,
-                **call_args
-            )
-            
-            processed_tensor = get_value_at_index(result_tuple, 0)
-            processed_np = (processed_tensor.squeeze(0).cpu().numpy().clip(0, 1) * 255.0).astype(np.uint8)
-            processed_pil_images.append(Image.fromarray(processed_np))
-            
-        return processed_pil_images
-
-    def run(self, input_type, image_input, video_input, preprocessor_name, model_name, zero_gpu_duration, *args, progress=gr.Progress(track_tqdm=True)):
-        from utils import app_utils
-        pil_images, is_video, fps = [], False, 30
-        
-        progress(0, desc="Reading input file...")
-        if input_type == "Image":
-            if image_input is None: raise gr.Error("Please provide an input image.")
-            pil_images = [image_input]
-        elif input_type == "Video":
-            if video_input is None: raise gr.Error("Please provide an input video.")
-            try:
-                video_reader = imageio.get_reader(video_input)
-                meta = video_reader.get_meta_data()
-                fps = meta.get('fps', 30)
-                pil_images = [Image.fromarray(frame) for frame in video_reader]
-                is_video = True
-                video_reader.close()
-            except Exception as e: raise gr.Error(f"Failed to read video file: {e}")
-        else:
-            raise gr.Error("Invalid input type selected.")
-
-        if not pil_images: raise gr.Error("Could not extract any frames from the input.")
-        
-        if app_utils.PREPROCESSOR_PARAMETER_MAP is None:
-            raise RuntimeError("Preprocessor parameter map is not built. Check startup logs.")
-            
-        params_config = app_utils.PREPROCESSOR_PARAMETER_MAP.get(preprocessor_name, [])
-        sliders_params = [p for p in params_config if p['type'] in ["INT", "FLOAT"]]
-        dropdown_params = [p for p in params_config if isinstance(p['type'], list)]
-        checkbox_params = [p for p in params_config if p['type'] == "BOOLEAN"]
-        ordered_params_config = sliders_params + dropdown_params + checkbox_params
-        param_names = [p['name'] for p in ordered_params_config]
-        provided_params = {param_names[i]: args[i] for i in range(len(param_names))}
-        
-        if preprocessor_name not in CPU_ONLY_PREPROCESSORS:
-            print(f"--- '{preprocessor_name}' requires GPU, requesting ZeroGPU. ---")
-            try:
-                processed_pil_images = self._execute_gpu_logic(
-                    self._gpu_logic,
-                    duration=zero_gpu_duration,
-                    default_duration=60,
-                    task_name=f"Preprocessor '{preprocessor_name}'",
-                    pil_images=pil_images,
-                    preprocessor_name=preprocessor_name,
-                    model_name=model_name,
-                    params=provided_params,
-                    progress=progress
-                )
-            except Exception as e:
-                import traceback; traceback.print_exc()
-                raise gr.Error(f"Failed to run preprocessor '{preprocessor_name}' on GPU: {e}")
-        else:
-            print(f"--- Running '{preprocessor_name}' on CPU, no ZeroGPU requested. ---")
-            try:
-                processed_pil_images = self._gpu_logic(pil_images, preprocessor_name, model_name, provided_params, progress=progress)
-            except Exception as e:
-                import traceback; traceback.print_exc()
-                raise gr.Error(f"Failed to run preprocessor '{preprocessor_name}' on CPU: {e}")
-
-        if not processed_pil_images: raise gr.Error("Processing returned no frames.")
-        
-        progress(0.9, desc="Finalizing output...")
-        if is_video:
-            frames_np = [np.array(img) for img in processed_pil_images]
-            frames_tensor = torch.from_numpy(np.stack(frames_np)).to(torch.float32) / 255.0
-            video_path = self._encode_video_from_frames(frames_tensor, fps, progress)
-            return [video_path]
-        else:
-            progress(1.0, desc="Done!")
-            return processed_pil_images

core/pipelines/sd_image_pipeline.py

@@ -11,12 +11,20 @@ import numpy as np
 from .base_pipeline import BasePipeline
 from core.settings import *
 from comfy_integration.nodes import *
-from utils.app_utils import get_value_at_index, sanitize_prompt, get_lora_path, get_embedding_path, ensure_controlnet_model_downloaded, sanitize_filename
+from utils.app_utils import get_value_at_index, sanitize_prompt, get_lora_path, get_embedding_path, ensure_controlnet_model_downloaded, ensure_ipadapter_models_downloaded, sanitize_filename
 from core.workflow_assembler import WorkflowAssembler
 
 class SdImagePipeline(BasePipeline):
     def get_required_models(self, model_display_name: str, **kwargs) -> List[str]:
-        return [model_display_name]
+        model_info = ALL_MODEL_MAP.get(model_display_name)
+        if not model_info:
+            return [model_display_name]
+
+        path_or_components = model_info[1]
+        if isinstance(path_or_components, dict):
+            return [v for v in path_or_components.values() if v and v != "pixel_space"]
+        else:
+            return [model_display_name]
 
     def _topological_sort(self, workflow: Dict[str, Any]) -> List[str]:
         graph = defaultdict(list)
@@ -47,7 +55,6 @@ class SdImagePipeline(BasePipeline):
             
         return sorted_nodes
 
-
     def _execute_workflow(self, workflow: Dict[str, Any], initial_objects: Dict[str, Any]):
         with torch.no_grad():
             computed_outputs = initial_objects
@@ -119,7 +126,7 @@ class SdImagePipeline(BasePipeline):
         progress(0.4, desc="Executing workflow...")
         
         initial_objects = {}
-        
+
         decoded_images_tensor = self._execute_workflow(workflow, initial_objects=initial_objects)
         
         output_images = []
@@ -135,6 +142,7 @@ class SdImagePipeline(BasePipeline):
             params_string = f"{ui_inputs['positive_prompt']}\nNegative prompt: {ui_inputs['negative_prompt']}\n"
             params_string += f"Steps: {ui_inputs['num_inference_steps']}, Sampler: {ui_inputs['sampler']}, Scheduler: {ui_inputs['scheduler']}, CFG scale: {ui_inputs['guidance_scale']}, Seed: {current_seed}, Size: {width_for_meta}x{height_for_meta}, Base Model: {model_display_name}"
             if ui_inputs['task_type'] != 'txt2img': params_string += f", Denoise: {ui_inputs['denoise']}"
+            if ui_inputs.get('clip_skip') and ui_inputs['clip_skip'] != 1: params_string += f", Clip skip: {abs(ui_inputs['clip_skip'])}"
             if loras_string: params_string += f", {loras_string}"
             
             pil_image.info = {'parameters': params_string.strip()}
@@ -146,39 +154,46 @@ class SdImagePipeline(BasePipeline):
         progress(0, desc="Preparing models...")
         
         task_type = ui_inputs['task_type']
+        model_display_name = ui_inputs['model_display_name']
+        model_type = MODEL_TYPE_MAP.get(model_display_name, 'sdxl')
+        
+        architectures_dict = ARCHITECTURES_CONFIG.get('architectures', {})
+        workflow_model_type = architectures_dict.get(model_type, {}).get("model_type", "sdxl")
         
         ui_inputs['positive_prompt'] = sanitize_prompt(ui_inputs.get('positive_prompt', ''))
         ui_inputs['negative_prompt'] = sanitize_prompt(ui_inputs.get('negative_prompt', ''))
         
-        required_models = self.get_required_models(model_display_name=ui_inputs['model_display_name'])
-        
+        if 'clip_skip' in ui_inputs and ui_inputs['clip_skip'] is not None:
+             ui_inputs['clip_skip'] = -int(ui_inputs['clip_skip'])
+        else:
+             ui_inputs['clip_skip'] = -1
+
+        required_models = self.get_required_models(model_display_name=model_display_name)
         self.model_manager.ensure_models_downloaded(required_models, progress=progress)
         
         lora_data = ui_inputs.get('lora_data', [])
         active_loras_for_gpu, active_loras_for_meta = [], []
         if lora_data:
             sources, ids, scales, files = lora_data[0::4], lora_data[1::4], lora_data[2::4], lora_data[3::4]
-            
             for i, (source, lora_id, scale, _) in enumerate(zip(sources, ids, scales, files)):
                 if scale > 0 and lora_id and lora_id.strip():
                     lora_filename = None
                     if source == "File":
                         lora_filename = sanitize_filename(lora_id)
                     elif source == "Civitai":
-                        local_path, status = get_lora_path(source, lora_id, ui_inputs['civitai_api_key'], progress)
+                        local_path, status = get_lora_path(source, lora_id, os.environ.get("CIVITAI_API_KEY", ""), progress)
                         if local_path: lora_filename = os.path.basename(local_path)
                         else: raise gr.Error(f"Failed to prepare LoRA {lora_id}: {status}")
                     
                     if lora_filename:
                         active_loras_for_gpu.append({"lora_name": lora_filename, "strength_model": scale, "strength_clip": scale})
                         active_loras_for_meta.append(f"{source} {lora_id}:{scale}")
-        
+
         ui_inputs['denoise'] = 1.0
         if task_type == 'img2img': ui_inputs['denoise'] = ui_inputs.get('img2img_denoise', 0.7)
         elif task_type == 'hires_fix': ui_inputs['denoise'] = ui_inputs.get('hires_denoise', 0.55)
         
         temp_files_to_clean = []
-        
         if not os.path.exists(INPUT_DIR): os.makedirs(INPUT_DIR)
 
         if task_type == 'img2img':
@@ -197,7 +212,6 @@ class SdImagePipeline(BasePipeline):
                 raise gr.Error("Inpainting requires an input image and a drawn mask.")
             
             background_img = inpaint_dict['background'].convert("RGBA")
-            
             composite_mask_pil = Image.new('L', background_img.size, 0)
             for layer in inpaint_dict['layers']:
                 if layer:
@@ -211,7 +225,7 @@ class SdImagePipeline(BasePipeline):
             temp_file_path = os.path.join(INPUT_DIR, f"temp_inpaint_composite_{random.randint(1000, 9999)}.png")
             composite_image_with_mask.save(temp_file_path, "PNG")
             
-            ui_inputs['inpaint_image'] = os.path.basename(temp_file_path)
+            ui_inputs['input_image'] = os.path.basename(temp_file_path)
             temp_files_to_clean.append(temp_file_path)
             ui_inputs.pop('inpaint_mask', None)
 
@@ -222,6 +236,9 @@ class SdImagePipeline(BasePipeline):
                 input_image_pil.save(temp_file_path, "PNG")
                 ui_inputs['input_image'] = os.path.basename(temp_file_path)
                 temp_files_to_clean.append(temp_file_path)
+            
+            ui_inputs['megapixels'] = 0.25
+            ui_inputs['grow_mask_by'] = ui_inputs.get('feathering', 10)
 
         elif task_type == 'hires_fix':
             input_image_pil = ui_inputs.get('hires_image')
@@ -241,7 +258,7 @@ class SdImagePipeline(BasePipeline):
                     if source == "File":
                         emb_filename = sanitize_filename(emb_id)
                     elif source == "Civitai":
-                        local_path, status = get_embedding_path(source, emb_id, ui_inputs['civitai_api_key'], progress)
+                        local_path, status = get_embedding_path(source, emb_id, os.environ.get("CIVITAI_API_KEY", ""), progress)
                         if local_path: emb_filename = os.path.basename(local_path)
                         else: raise gr.Error(f"Failed to prepare Embedding {emb_id}: {status}")
                     
@@ -255,20 +272,162 @@ class SdImagePipeline(BasePipeline):
             else:
                 ui_inputs['positive_prompt'] = embedding_prompt_text
 
+        controlnet_data = ui_inputs.get('controlnet_data', [])
+        active_controlnets = []
+        if controlnet_data:
+            (cn_images, _, _, cn_strengths, cn_filepaths) = [controlnet_data[i::5] for i in range(5)]
+            for i in range(len(cn_images)):
+                if cn_images[i] and cn_strengths[i] > 0 and cn_filepaths[i] and cn_filepaths[i] != "None":
+                    ensure_controlnet_model_downloaded(cn_filepaths[i], progress)
+                    if not os.path.exists(INPUT_DIR): os.makedirs(INPUT_DIR)
+                    cn_temp_path = os.path.join(INPUT_DIR, f"temp_cn_{i}_{random.randint(1000, 9999)}.png")
+                    cn_images[i].save(cn_temp_path, "PNG")
+                    temp_files_to_clean.append(cn_temp_path)
+                    active_controlnets.append({
+                        "image": os.path.basename(cn_temp_path), "strength": cn_strengths[i],
+                        "start_percent": 0.0, "end_percent": 1.0, "control_net_name": cn_filepaths[i]
+                    })
+                    
+        diffsynth_controlnet_data = ui_inputs.get('diffsynth_controlnet_data', [])
+        active_diffsynth_controlnets = []
+        if diffsynth_controlnet_data:
+            (cn_images, _, _, cn_strengths, cn_filepaths) = [diffsynth_controlnet_data[i::5] for i in range(5)]
+            for i in range(len(cn_images)):
+                if cn_images[i] and cn_strengths[i] > 0 and cn_filepaths[i] and cn_filepaths[i] != "None":
+                    ensure_controlnet_model_downloaded(cn_filepaths[i], progress)
+
+                    if not os.path.exists(INPUT_DIR): os.makedirs(INPUT_DIR)
+                    cn_temp_path = os.path.join(INPUT_DIR, f"temp_diffsynth_cn_{i}_{random.randint(1000, 9999)}.png")
+                    cn_images[i].save(cn_temp_path, "PNG")
+                    temp_files_to_clean.append(cn_temp_path)
+                    active_diffsynth_controlnets.append({
+                        "image": os.path.basename(cn_temp_path), "strength": cn_strengths[i],
+                        "control_net_name": cn_filepaths[i]
+                    })
+        
+        ipadapter_data = ui_inputs.get('ipadapter_data', [])
+        active_ipadapters = []
+        if ipadapter_data:
+            num_ipa_units = (len(ipadapter_data) - 5) // 3
+            final_preset, final_weight, final_lora_strength, final_embeds_scaling, final_combine_method = ipadapter_data[-5:]
+            ipa_images, ipa_weights, ipa_lora_strengths = [ipadapter_data[i*num_ipa_units:(i+1)*num_ipa_units] for i in range(3)]
+            all_presets_to_download = set()
+            for i in range(num_ipa_units):
+                if ipa_images[i] and ipa_weights[i] > 0 and final_preset:
+                    all_presets_to_download.add(final_preset)
+                    if not os.path.exists(INPUT_DIR): os.makedirs(INPUT_DIR)
+                    ipa_temp_path = os.path.join(INPUT_DIR, f"temp_ipa_{i}_{random.randint(1000, 9999)}.png")
+                    ipa_images[i].save(ipa_temp_path, "PNG")
+                    temp_files_to_clean.append(ipa_temp_path)
+                    active_ipadapters.append({
+                        "image": os.path.basename(ipa_temp_path), "preset": final_preset,
+                        "weight": ipa_weights[i], "lora_strength": ipa_lora_strengths[i]
+                    })
+            if active_ipadapters and final_preset:
+                all_presets_to_download.add(final_preset)
+            for preset in all_presets_to_download:
+                ensure_ipadapter_models_downloaded(preset, progress)
+                
+            model_type_key = 'sd15' if workflow_model_type == 'sd15' else 'sdxl'
+            if active_ipadapters:
+                active_ipadapters.append({
+                    'is_final_settings': True, 'model_type': model_type_key, 'final_preset': final_preset,
+                    'final_weight': final_weight, 'final_lora_strength': final_lora_strength,
+                    'final_embeds_scaling': final_embeds_scaling, 'final_combine_method': final_combine_method
+                })
+                
+        flux1_ipadapter_data = ui_inputs.get('flux1_ipadapter_data', [])
+        active_flux1_ipadapters = []
+        if flux1_ipadapter_data:
+            num_units = len(flux1_ipadapter_data) // 4
+            f_images = flux1_ipadapter_data[0*num_units : 1*num_units]
+            f_weights = flux1_ipadapter_data[1*num_units : 2*num_units]
+            f_starts = flux1_ipadapter_data[2*num_units : 3*num_units]
+            f_ends = flux1_ipadapter_data[3*num_units : 4*num_units]
+            for i in range(len(f_images)):
+                if f_images[i] and f_weights[i] > 0:
+                    from utils.app_utils import _ensure_model_downloaded
+                    for filename in ["ip-adapter.bin"]:
+                        _ensure_model_downloaded(filename, progress)
+                    
+                    from huggingface_hub import snapshot_download
+                    progress(0.5, desc="Caching HF SigLIP model...")
+                    snapshot_download(
+                        repo_id="google/siglip-so400m-patch14-384", 
+                        allow_patterns=["*.json", "*.safetensors", "*.txt"], 
+                        ignore_patterns=["*.msgpack", "*.h5", "*.bin"]
+                    )
+
+                    temp_path = os.path.join(INPUT_DIR, f"temp_fipa_{i}_{random.randint(1000, 9999)}.png")
+                    f_images[i].save(temp_path, "PNG")
+                    temp_files_to_clean.append(temp_path)
+                    active_flux1_ipadapters.append({
+                        "image": os.path.basename(temp_path),
+                        "weight": f_weights[i], "start_percent": f_starts[i], "end_percent": f_ends[i]
+                    })
+                    
+        sd3_ipadapter_data = ui_inputs.get('sd3_ipadapter_chain', [])
+        active_sd3_ipadapters = []
+        if sd3_ipadapter_data:
+            num_units = len(sd3_ipadapter_data) // 4
+            s_images = sd3_ipadapter_data[0*num_units : 1*num_units]
+            s_weights = sd3_ipadapter_data[1*num_units : 2*num_units]
+            s_starts = sd3_ipadapter_data[2*num_units : 3*num_units]
+            s_ends = sd3_ipadapter_data[3*num_units : 4*num_units]
+            sd3_ipa_downloaded = False
+            for i in range(len(s_images)):
+                if s_images[i] and s_weights[i] > 0:
+                    if not sd3_ipa_downloaded:
+                        from utils.app_utils import ensure_sd3_ipadapter_models_downloaded
+                        ensure_sd3_ipadapter_models_downloaded(progress)
+                        sd3_ipa_downloaded = True
+                    temp_path = os.path.join(INPUT_DIR, f"temp_s3ipa_{i}_{random.randint(1000, 9999)}.png")
+                    s_images[i].save(temp_path, "PNG")
+                    temp_files_to_clean.append(temp_path)
+                    active_sd3_ipadapters.append({
+                        "image": os.path.basename(temp_path),
+                        "weight": s_weights[i], "start_percent": s_starts[i], "end_percent": s_ends[i]
+                    })
+                    
+        style_data = ui_inputs.get('style_data', [])
+        active_styles = []
+        if style_data:
+            num_units = len(style_data) // 2
+            st_images = style_data[0*num_units : 1*num_units]
+            st_strengths = style_data[1*num_units : 2*num_units]
+            for i in range(len(st_images)):
+                if st_images[i] and st_strengths[i] > 0:
+                    from utils.app_utils import _ensure_model_downloaded
+                    _ensure_model_downloaded("sigclip_vision_patch14_384.safetensors", progress)
+                    temp_path = os.path.join(INPUT_DIR, f"temp_style_{i}_{random.randint(1000, 9999)}.png")
+                    st_images[i].save(temp_path, "PNG")
+                    temp_files_to_clean.append(temp_path)
+                    active_styles.append({
+                        "image": os.path.basename(temp_path), "strength": st_strengths[i]
+                    })
+        
+        reference_latent_data = ui_inputs.get('reference_latent_data', [])
+        active_reference_latents = []
+        if reference_latent_data:
+            for img in reference_latent_data:
+                if img:
+                    if not os.path.exists(INPUT_DIR): os.makedirs(INPUT_DIR)
+                    temp_path = os.path.join(INPUT_DIR, f"temp_ref_{random.randint(1000, 9999)}.png")
+                    img.save(temp_path, "PNG")
+                    temp_files_to_clean.append(temp_path)
+                    active_reference_latents.append(os.path.basename(temp_path))
+        
         from utils.app_utils import get_vae_path
         vae_source = ui_inputs.get('vae_source')
         vae_id = ui_inputs.get('vae_id')
-        vae_file = ui_inputs.get('vae_file')
         vae_name_override = None
-
         if vae_source and vae_source != "None":
             if vae_source == "File":
                 vae_name_override = sanitize_filename(vae_id)
             elif vae_source == "Civitai" and vae_id and vae_id.strip():
-                local_path, status = get_vae_path(vae_source, vae_id, ui_inputs.get('civitai_api_key'), progress)
+                local_path, status = get_vae_path(vae_source, vae_id, os.environ.get("CIVITAI_API_KEY", ""), progress)
                 if local_path: vae_name_override = os.path.basename(local_path)
                 else: raise gr.Error(f"Failed to prepare VAE {vae_id}: {status}")
-        
         if vae_name_override:
             ui_inputs['vae_name'] = vae_name_override
 
@@ -276,78 +435,84 @@ class SdImagePipeline(BasePipeline):
         active_conditioning = []
         if conditioning_data:
             num_units = len(conditioning_data) // 6
-            prompts   = conditioning_data[0*num_units : 1*num_units]
-            widths    = conditioning_data[1*num_units : 2*num_units]
-            heights   = conditioning_data[2*num_units : 3*num_units]
-            xs        = conditioning_data[3*num_units : 4*num_units]
-            ys        = conditioning_data[4*num_units : 5*num_units]
-            strengths = conditioning_data[5*num_units : 6*num_units]
-
+            prompts, widths, heights, xs, ys, strengths = [conditioning_data[i*num_units : (i+1)*num_units] for i in range(6)]
             for i in range(num_units):
                 if prompts[i] and prompts[i].strip():
                     active_conditioning.append({
-                        "prompt": prompts[i],
-                        "width": int(widths[i]),
-                        "height": int(heights[i]),
-                        "x": int(xs[i]),
-                        "y": int(ys[i]),
-                        "strength": float(strengths[i])
+                        "prompt": prompts[i], "width": int(widths[i]), "height": int(heights[i]),
+                        "x": int(xs[i]), "y": int(ys[i]), "strength": float(strengths[i])
                     })
 
-        reference_latent_data = ui_inputs.get('reference_latent_data', [])
-        active_reference_latents = []
-        if reference_latent_data:
-            for img_pil in reference_latent_data:
-                if img_pil is not None:
-                    temp_file_path = os.path.join(INPUT_DIR, f"temp_ref_{random.randint(1000, 9999)}.png")
-                    img_pil.save(temp_file_path, "PNG")
-                    active_reference_latents.append(os.path.basename(temp_file_path))
-                    temp_files_to_clean.append(temp_file_path)
-
         loras_string = f"LoRAs: [{', '.join(active_loras_for_meta)}]" if active_loras_for_meta else ""
         
         progress(0.8, desc="Assembling workflow...")
         
         if ui_inputs.get('seed') == -1:
             ui_inputs['seed'] = random.randint(0, 2**32 - 1)
+        
+        model_info = ALL_MODEL_MAP[model_display_name]
+        path_or_components = model_info[1]
+        latent_type = model_info[3] if len(model_info) > 3 and model_info[3] else 'latent'
+        latent_generator_template = "EmptyLatentImage"
+        if latent_type == 'sd3_latent':
+            latent_generator_template = "EmptySD3LatentImage"
+        elif latent_type == 'chroma_radiance_latent':
+            latent_generator_template = "EmptyChromaRadianceLatentImage"
+        elif latent_type == 'hunyuan_latent':
+            latent_generator_template = "EmptyHunyuanImageLatent"
 
-        dynamic_values = {'task_type': ui_inputs['task_type']}
+        dynamic_values = {
+            'task_type': ui_inputs['task_type'], 
+            'model_type': workflow_model_type,
+            'latent_type': latent_type,
+            'latent_generator_template': latent_generator_template
+        }
         
         recipe_path = os.path.join(os.path.dirname(__file__), "workflow_recipes", "sd_unified_recipe.yaml")
         assembler = WorkflowAssembler(recipe_path, dynamic_values=dynamic_values)
         
-        model_display_name = ui_inputs['model_display_name']
-        if model_display_name not in ALL_MODEL_MAP:
-            raise gr.Error(f"Model '{model_display_name}' is not configured in model_list.yaml.")
-        
-        _, components, _, _ = ALL_MODEL_MAP[model_display_name]
-        
         workflow_inputs = {
+            **ui_inputs,
             "positive_prompt": ui_inputs['positive_prompt'], "negative_prompt": ui_inputs['negative_prompt'],
             "seed": ui_inputs['seed'], "steps": ui_inputs['num_inference_steps'], "cfg": ui_inputs['guidance_scale'],
             "sampler_name": ui_inputs['sampler'], "scheduler": ui_inputs['scheduler'],
             "batch_size": ui_inputs['batch_size'],
-            "denoise": ui_inputs['denoise'], 
-            "input_image": ui_inputs.get('input_image'),
-            "inpaint_image": ui_inputs.get('inpaint_image'),
-            "inpaint_mask": ui_inputs.get('inpaint_mask'),
-            "left": ui_inputs.get('outpaint_left'), "top": ui_inputs.get('outpaint_top'),
-            "right": ui_inputs.get('outpaint_right'), "bottom": ui_inputs.get('outpaint_bottom'),
-            "hires_upscaler": ui_inputs.get('hires_upscaler'), "hires_scale_by": ui_inputs.get('hires_scale_by'),
-            "unet_name": components['unet'],
-            "clip_name": components['clip'],
-            "vae_name": ui_inputs.get('vae_name', components['vae']),
+            "clip_skip": ui_inputs['clip_skip'],
+            "denoise": ui_inputs['denoise'],
+            "vae_name": ui_inputs.get('vae_name'),
+            "guidance": ui_inputs.get('guidance', 3.5),
             "lora_chain": active_loras_for_gpu,
+            "controlnet_chain": active_controlnets,
+            "diffsynth_controlnet_chain": active_diffsynth_controlnets,
+            "ipadapter_chain": active_ipadapters,
+            "flux1_ipadapter_chain": active_flux1_ipadapters,
+            "sd3_ipadapter_chain": active_sd3_ipadapters,
+            "style_chain": active_styles,
             "conditioning_chain": active_conditioning,
             "reference_latent_chain": active_reference_latents,
+            "vae_chain": [ui_inputs.get('vae_name')] if ui_inputs.get('vae_name') else [],
         }
+
+        if isinstance(path_or_components, dict):
+            workflow_inputs.update({
+                'unet_name': path_or_components.get('unet'),
+                'vae_name': ui_inputs.get('vae_name') or path_or_components.get('vae'),
+                'clip_name': path_or_components.get('clip'),
+                'clip1_name': path_or_components.get('clip1'),
+                'clip2_name': path_or_components.get('clip2'),
+                'clip3_name': path_or_components.get('clip3'),
+                'clip4_name': path_or_components.get('clip4'),
+                'lora_name': path_or_components.get('lora'),
+            })
+        else:
+            workflow_inputs['model_name'] = path_or_components
         
         if task_type == 'txt2img':
             workflow_inputs['width'] = ui_inputs['width']
             workflow_inputs['height'] = ui_inputs['height']
         
         workflow = assembler.assemble(workflow_inputs)
-        
+
         progress(1.0, desc="All models ready. Requesting GPU for generation...")
 
         try:
@@ -362,7 +527,7 @@ class SdImagePipeline(BasePipeline):
                 assembler=assembler, 
                 progress=progress
             )
-
+            
             import json
             import glob
             from PIL import PngImagePlugin

core/pipelines/workflow_recipes/_partials/{_base_sampler.yaml → _base_sampler_sd.yaml}

@@ -1,11 +1,21 @@
 nodes:
+  pos_prompt:
+    class_type: CLIPTextEncode
+    title: "CLIP Text Encode (Positive)"
+  neg_prompt:
+    class_type: CLIPTextEncode
+    title: "CLIP Text Encode (Negative)"
   ksampler:
     class_type: KSampler
-
+    title: "KSampler"
+    params:
+      denoise: 1.0
   vae_decode:
     class_type: VAEDecode
+    title: "VAE Decode"
   save_image:
     class_type: SaveImage
+    title: "Save Image"
     params: {}
 
 connections:
@@ -15,9 +25,12 @@ connections:
     to: "save_image:images"
 
 ui_map:
+  positive_prompt: "pos_prompt:text"
+  negative_prompt: "neg_prompt:text"
   seed: "ksampler:seed"
   steps: "ksampler:steps"
   cfg: "ksampler:cfg"
   sampler_name: "ksampler:sampler_name"
   scheduler: "ksampler:scheduler"
-  denoise: "ksampler:denoise"
+  denoise: "ksampler:denoise"
+  filename_prefix: "save_image:filename_prefix"

core/pipelines/workflow_recipes/_partials/conditioning/anima.yaml

@@ -0,0 +1,54 @@
+nodes:
+  unet_loader:
+    class_type: UNETLoader
+    title: "Load Diffusion Model"
+    params:
+      weight_dtype: "default"
+  vae_loader:
+    class_type: VAELoader
+    title: "Load VAE"
+  clip_loader:
+    class_type: CLIPLoader
+    title: "Load CLIP"
+    params:
+      type: "stable_diffusion"
+      device: "default"
+
+connections:
+  - from: "unet_loader:0"
+    to: "ksampler:model"
+  - from: "clip_loader:0"
+    to: "pos_prompt:clip"
+  - from: "clip_loader:0"
+    to: "neg_prompt:clip"
+  - from: "vae_loader:0"
+    to: "vae_decode:vae"
+  - from: "vae_loader:0"
+    to: "vae_encode:vae"
+  - from: "pos_prompt:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+
+dynamic_lora_chains:
+  lora_chain:
+    template: "LoraLoader"
+    output_map:
+      "unet_loader:0": "model"
+      "clip_loader:0": "clip"
+    input_map:
+      "model": "model"
+      "clip": "clip"
+    end_input_map:
+      "model": ["ksampler:model"]
+      "clip": ["pos_prompt:clip", "neg_prompt:clip"]
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    ksampler_node: "ksampler"
+    clip_source: "clip_loader:0"
+
+ui_map:
+  unet_name: "unet_loader:unet_name"
+  vae_name: "vae_loader:vae_name"
+  clip_name: "clip_loader:clip_name"

core/pipelines/workflow_recipes/_partials/conditioning/chroma1-radiance.yaml

@@ -0,0 +1,59 @@
+nodes:
+  unet_loader:
+    class_type: UNETLoader
+    title: "Load Diffusion Model"
+    params:
+      weight_dtype: "default"
+  vae_loader:
+    class_type: VAELoader
+    title: "Load VAE"
+    params:
+      vae_name: "pixel_space"
+  clip_loader:
+    class_type: CLIPLoader
+    title: "Load CLIP"
+    params:
+      type: "chroma"
+      device: "default"
+  t5_tokenizer:
+    class_type: T5TokenizerOptions
+    title: "T5TokenizerOptions"
+    params:
+      min_padding: 0
+      min_length: 3
+  model_sampler:
+    class_type: ModelSamplingAuraFlow
+    params:
+      shift: 3.0
+
+connections:
+  - from: "unet_loader:0"
+    to: "model_sampler:model"
+  - from: "model_sampler:0"
+    to: "ksampler:model"
+  
+  - from: "clip_loader:0"
+    to: "t5_tokenizer:clip"
+  - from: "t5_tokenizer:0"
+    to: "pos_prompt:clip"
+  - from: "t5_tokenizer:0"
+    to: "neg_prompt:clip"
+    
+  - from: "pos_prompt:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+  
+  - from: "vae_loader:0"
+    to: "vae_decode:vae"
+  - from: "vae_loader:0"
+    to: "vae_encode:vae"
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    ksampler_node: "ksampler"
+    clip_source: "t5_tokenizer:0"
+
+ui_map:
+  unet_name: "unet_loader:unet_name"
+  clip_name: "clip_loader:clip_name"

core/pipelines/workflow_recipes/_partials/conditioning/chroma1.yaml

@@ -0,0 +1,61 @@
+nodes:
+  unet_loader:
+    class_type: UNETLoader
+    title: "Load Diffusion Model"
+    params:
+      weight_dtype: "default"
+  vae_loader:
+    class_type: VAELoader
+    title: "Load VAE"
+  clip_loader:
+    class_type: CLIPLoader
+    title: "Load CLIP"
+    params:
+      type: "chroma"
+      device: "default"
+  t5_tokenizer:
+    class_type: T5TokenizerOptions
+    title: "T5TokenizerOptions"
+    params:
+      min_padding: 1
+      min_length: 0
+  fresca:
+    class_type: FreSca
+    title: "FreSca"
+    params:
+      scale_low: 1.0
+      scale_high: 2.5
+      freq_cutoff: 30
+
+connections:
+  - from: "unet_loader:0"
+    to: "fresca:model"
+  - from: "fresca:0"
+    to: "ksampler:model"
+  
+  - from: "clip_loader:0"
+    to: "t5_tokenizer:clip"
+  - from: "t5_tokenizer:0"
+    to: "pos_prompt:clip"
+  - from: "t5_tokenizer:0"
+    to: "neg_prompt:clip"
+    
+  - from: "pos_prompt:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+  
+  - from: "vae_loader:0"
+    to: "vae_decode:vae"
+  - from: "vae_loader:0"
+    to: "vae_encode:vae"
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    ksampler_node: "ksampler"
+    clip_source: "t5_tokenizer:0"
+
+ui_map:
+  unet_name: "unet_loader:unet_name"
+  vae_name: "vae_loader:vae_name"
+  clip_name: "clip_loader:clip_name"

core/pipelines/workflow_recipes/_partials/conditioning/ernie-image.yaml

@@ -0,0 +1,54 @@
+nodes:
+  unet_loader:
+    class_type: UNETLoader
+    title: "Load Diffusion Model"
+    params:
+      weight_dtype: "default"
+  clip_loader:
+    class_type: CLIPLoader
+    title: "Load CLIP"
+    params:
+      type: "flux2"
+      device: "default"
+  vae_loader:
+    class_type: VAELoader
+    title: "Load VAE"
+
+connections:
+  - from: "unet_loader:0"
+    to: "ksampler:model"
+  - from: "clip_loader:0"
+    to: "pos_prompt:clip"
+  - from: "clip_loader:0"
+    to: "neg_prompt:clip"
+  - from: "pos_prompt:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+  - from: "vae_loader:0"
+    to: "vae_decode:vae"
+  - from: "vae_loader:0"
+    to: "vae_encode:vae"
+
+dynamic_lora_chains:
+  lora_chain:
+    template: "LoraLoader"
+    output_map:
+      "unet_loader:0": "model"
+      "clip_loader:0": "clip"
+    input_map:
+      "model": "model"
+      "clip": "clip"
+    end_input_map:
+      "model": ["ksampler:model"]
+      "clip": ["pos_prompt:clip", "neg_prompt:clip"]
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    ksampler_node: "ksampler"
+    clip_source: "clip_loader:0"
+
+ui_map:
+  unet_name: "unet_loader:unet_name"
+  clip_name: "clip_loader:clip_name"
+  vae_name: "vae_loader:vae_name"

core/pipelines/workflow_recipes/_partials/conditioning/flux1.yaml

@@ -0,0 +1,64 @@
+nodes:
+  unet_loader:
+    class_type: UNETLoader
+    title: "Load FLUX UNET"
+    params:
+      weight_dtype: "default"
+  vae_loader:
+    class_type: VAELoader
+    title: "Load FLUX VAE"
+  clip_loader:
+    class_type: DualCLIPLoader
+    title: "Load FLUX Dual CLIP"
+    params:
+      type: "flux"
+      device: "default"
+  flux_guidance:
+    class_type: FluxGuidance
+    title: "FluxGuidance"
+
+connections:
+  - from: "unet_loader:0"
+    to: "ksampler:model"
+  - from: "clip_loader:0"
+    to: "pos_prompt:clip"
+  - from: "clip_loader:0"
+    to: "neg_prompt:clip"
+  - from: "vae_loader:0"
+    to: "vae_decode:vae"
+  - from: "vae_loader:0"
+    to: "vae_encode:vae"
+  - from: "pos_prompt:0"
+    to: "flux_guidance:conditioning"
+  - from: "flux_guidance:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+
+dynamic_controlnet_chains:
+  controlnet_chain:
+    template: "ControlNetApplyAdvanced"
+    ksampler_node: "ksampler"
+    vae_source: "vae_loader:0"
+
+dynamic_flux1_ipadapter_chains:
+  flux1_ipadapter_chain:
+    ksampler_node: "ksampler"
+
+dynamic_style_chains:
+  style_chain:
+    flux_guidance_node: "flux_guidance"
+    ksampler_node: "ksampler"
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    flux_guidance_node: "flux_guidance"
+    ksampler_node: "ksampler"
+    clip_source: "clip_loader:0"
+
+ui_map:
+  unet_name: "unet_loader:unet_name"
+  vae_name: "vae_loader:vae_name"
+  clip1_name: "clip_loader:clip_name1"
+  clip2_name: "clip_loader:clip_name2"
+  guidance: "flux_guidance:guidance"

core/pipelines/workflow_recipes/_partials/conditioning/flux2-kv.yaml

@@ -0,0 +1,104 @@
+nodes:
+  unet_loader:
+    class_type: UNETLoader
+    title: "Load Diffusion Model"
+    params:
+      weight_dtype: "default"
+  clip_loader:
+    class_type: CLIPLoader
+    title: "Load CLIP"
+    params:
+      type: "flux2"
+      device: "default"
+  vae_loader:
+    class_type: VAELoader
+    title: "Load VAE"
+    
+  flux_kv_cache:
+    class_type: FluxKVCache
+    title: "Flux KV Cache"
+  
+  pos_prompt:
+    class_type: CLIPTextEncode
+    title: "CLIP Text Encode (Positive)"
+  neg_prompt:
+    class_type: CLIPTextEncode
+    title: "CLIP Text Encode (Negative)"
+  
+  ksampler:
+    class_type: KSampler
+    title: "KSampler"
+    params:
+      denoise: 1.0
+
+  vae_decode:
+    class_type: VAEDecode
+    title: "VAE Decode"
+    
+  save_image:
+    class_type: SaveImage
+    title: "Save Image"
+
+connections:
+  - from: "unet_loader:0"
+    to: "flux_kv_cache:model"
+  - from: "flux_kv_cache:0"
+    to: "ksampler:model"
+    
+  - from: "clip_loader:0"
+    to: "pos_prompt:clip"
+  - from: "clip_loader:0"
+    to: "neg_prompt:clip"
+    
+  - from: "vae_loader:0"
+    to: "vae_decode:vae"
+  - from: "vae_loader:0"
+    to: "vae_encode:vae"
+  
+  - from: "pos_prompt:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+    
+  - from: "latent_source:0"
+    to: "ksampler:latent_image"
+    
+  - from: "ksampler:0"
+    to: "vae_decode:samples"
+  - from: "vae_decode:0"
+    to: "save_image:images"
+
+dynamic_lora_chains:
+  lora_chain:
+    template: "LoraLoader"
+    output_map:
+      "unet_loader:0": "model"
+      "clip_loader:0": "clip"
+    input_map:
+      "model": "model"
+      "clip": "clip"
+    end_input_map:
+      "model": ["flux_kv_cache:model"]
+      "clip": ["pos_prompt:clip", "neg_prompt:clip"]
+
+dynamic_reference_latent_chains:
+  reference_latent_chain:
+    ksampler_node: "ksampler"
+    vae_node: "vae_loader"
+
+ui_map:
+  unet_name: "unet_loader:unet_name"
+  clip_name: "clip_loader:clip_name"
+  vae_name: "vae_loader:vae_name"
+  
+  positive_prompt: "pos_prompt:text"
+  negative_prompt: "neg_prompt:text"
+  
+  seed: "ksampler:seed"
+  steps: "ksampler:steps"
+  cfg: "ksampler:cfg"
+  sampler_name: "ksampler:sampler_name"
+  scheduler: "ksampler:scheduler"
+  denoise: "ksampler:denoise"
+  
+  filename_prefix: "save_image:filename_prefix"

core/pipelines/workflow_recipes/_partials/conditioning/flux2.yaml

@@ -20,6 +20,20 @@ nodes:
   neg_prompt:
     class_type: CLIPTextEncode
     title: "CLIP Text Encode (Negative)"
+  
+  ksampler:
+    class_type: KSampler
+    title: "KSampler"
+    params:
+      denoise: 1.0
+
+  vae_decode:
+    class_type: VAEDecode
+    title: "VAE Decode"
+    
+  save_image:
+    class_type: SaveImage
+    title: "Save Image"
 
 connections:
   - from: "unet_loader:0"
@@ -37,6 +51,14 @@ connections:
     to: "ksampler:positive"
   - from: "neg_prompt:0"
     to: "ksampler:negative"
+    
+  - from: "latent_source:0"
+    to: "ksampler:latent_image"
+    
+  - from: "ksampler:0"
+    to: "vae_decode:samples"
+  - from: "vae_decode:0"
+    to: "save_image:images"
 
 dynamic_lora_chains:
   lora_chain:
@@ -51,11 +73,6 @@ dynamic_lora_chains:
       "model": ["ksampler:model"]
       "clip": ["pos_prompt:clip", "neg_prompt:clip"]
 
-dynamic_conditioning_chains:
-  conditioning_chain:
-    ksampler_node: "ksampler"
-    clip_source: "clip_loader:0"
-
 dynamic_reference_latent_chains:
   reference_latent_chain:
     ksampler_node: "ksampler"
@@ -65,5 +82,15 @@ ui_map:
   unet_name: "unet_loader:unet_name"
   clip_name: "clip_loader:clip_name"
   vae_name: "vae_loader:vae_name"
+  
   positive_prompt: "pos_prompt:text"
-  negative_prompt: "neg_prompt:text"
+  negative_prompt: "neg_prompt:text"
+  
+  seed: "ksampler:seed"
+  steps: "ksampler:steps"
+  cfg: "ksampler:cfg"
+  sampler_name: "ksampler:sampler_name"
+  scheduler: "ksampler:scheduler"
+  denoise: "ksampler:denoise"
+  
+  filename_prefix: "save_image:filename_prefix"

core/pipelines/workflow_recipes/_partials/conditioning/hidream.yaml

@@ -0,0 +1,53 @@
+nodes:
+  unet_loader:
+    class_type: UNETLoader
+    title: "Load HiDream UNET"
+    params:
+      weight_dtype: "default"
+  vae_loader:
+    class_type: VAELoader
+    title: "Load HiDream VAE"
+  clip_loader:
+    class_type: QuadrupleCLIPLoader
+    title: "Load HiDream Quadruple CLIP"
+  
+  model_sampler:
+    class_type: ModelSamplingSD3
+    title: "ModelSamplingSD3"
+    params:
+      shift: 6.0
+
+connections:
+  - from: "unet_loader:0"
+    to: "model_sampler:model"
+    
+  - from: "model_sampler:0"
+    to: "ksampler:model"
+    
+  - from: "clip_loader:0"
+    to: "pos_prompt:clip"
+  - from: "clip_loader:0"
+    to: "neg_prompt:clip"
+
+  - from: "pos_prompt:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+    
+  - from: "vae_loader:0"
+    to: "vae_decode:vae"
+  - from: "vae_loader:0"
+    to: "vae_encode:vae"
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    ksampler_node: "ksampler"
+    clip_source: "clip_loader:0"
+
+ui_map:
+  unet_name: "unet_loader:unet_name"
+  vae_name: "vae_loader:vae_name"
+  clip1_name: "clip_loader:clip_name1"
+  clip2_name: "clip_loader:clip_name2"
+  clip3_name: "clip_loader:clip_name3"
+  clip4_name: "clip_loader:clip_name4"

core/pipelines/workflow_recipes/_partials/conditioning/hunyuanimage.yaml

@@ -0,0 +1,42 @@
+nodes:
+  unet_loader:
+    class_type: UNETLoader
+    title: "Load Hunyuan UNET"
+    params:
+      weight_dtype: "default"
+  vae_loader:
+    class_type: VAELoader
+    title: "Load Hunyuan VAE"
+  clip_loader:
+    class_type: DualCLIPLoader
+    title: "Load Hunyuan Dual CLIP"
+    params:
+      type: "hunyuan_image"
+      device: "default"
+
+connections:
+  - from: "unet_loader:0"
+    to: "ksampler:model"
+  - from: "clip_loader:0"
+    to: "pos_prompt:clip"
+  - from: "clip_loader:0"
+    to: "neg_prompt:clip"
+  - from: "vae_loader:0"
+    to: "vae_decode:vae"
+  - from: "vae_loader:0"
+    to: "vae_encode:vae"
+  - from: "pos_prompt:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    ksampler_node: "ksampler"
+    clip_source: "clip_loader:0"
+
+ui_map:
+  unet_name: "unet_loader:unet_name"
+  vae_name: "vae_loader:vae_name"
+  clip1_name: "clip_loader:clip_name1"
+  clip2_name: "clip_loader:clip_name2"

core/pipelines/workflow_recipes/_partials/conditioning/longcat-image.yaml

@@ -0,0 +1,83 @@
+nodes:
+  unet_loader:
+    class_type: UNETLoader
+    title: "Load Diffusion Model"
+    params:
+      weight_dtype: "default"
+  vae_loader:
+    class_type: VAELoader
+    title: "Load VAE"
+  clip_loader:
+    class_type: CLIPLoader
+    title: "Load CLIP"
+    params:
+      type: "longcat_image"
+      device: "default"
+  
+  cfg_norm:
+    class_type: CFGNorm
+    title: "CFGNorm"
+    params:
+      strength: 1.0
+
+  flux_guidance_pos:
+    class_type: FluxGuidance
+    title: "FluxGuidance (Positive)"
+    params:
+      guidance: 4.0
+
+  flux_guidance_neg:
+    class_type: FluxGuidance
+    title: "FluxGuidance (Negative)"
+    params:
+      guidance: 4.0
+
+connections:
+  - from: "unet_loader:0"
+    to: "cfg_norm:model"
+  - from: "cfg_norm:0"
+    to: "ksampler:model"
+  
+  - from: "clip_loader:0"
+    to: "pos_prompt:clip"
+  - from: "clip_loader:0"
+    to: "neg_prompt:clip"
+    
+  - from: "pos_prompt:0"
+    to: "flux_guidance_pos:conditioning"
+  - from: "neg_prompt:0"
+    to: "flux_guidance_neg:conditioning"
+    
+  - from: "flux_guidance_pos:0"
+    to: "ksampler:positive"
+  - from: "flux_guidance_neg:0"
+    to: "ksampler:negative"
+  
+  - from: "vae_loader:0"
+    to: "vae_decode:vae"
+  - from: "vae_loader:0"
+    to: "vae_encode:vae"
+
+dynamic_lora_chains:
+  lora_chain:
+    template: "LoraLoader"
+    output_map:
+      "unet_loader:0": "model"
+      "clip_loader:0": "clip"
+    input_map:
+      "model": "model"
+      "clip": "clip"
+    end_input_map:
+      "model": ["cfg_norm:model"]
+      "clip": ["pos_prompt:clip", "neg_prompt:clip"]
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    flux_guidance_node: "flux_guidance_pos"
+    ksampler_node: "ksampler"
+    clip_source: "clip_loader:0"
+
+ui_map:
+  unet_name: "unet_loader:unet_name"
+  vae_name: "vae_loader:vae_name"
+  clip_name: "clip_loader:clip_name"

core/pipelines/workflow_recipes/_partials/conditioning/lumina.yaml

@@ -0,0 +1,57 @@
+nodes:
+  ckpt_loader:
+    class_type: CheckpointLoaderSimple
+    title: "Load Checkpoint"
+  model_sampler:
+    class_type: ModelSamplingAuraFlow
+    title: "ModelSamplingAuraFlow"
+    params:
+      shift: 4.0
+
+connections:
+  - from: "ckpt_loader:0"
+    to: "model_sampler:model"
+  - from: "model_sampler:0"
+    to: "ksampler:model"
+  
+  - from: "ckpt_loader:1"
+    to: "pos_prompt:clip"
+  - from: "ckpt_loader:1"
+    to: "neg_prompt:clip"
+  - from: "pos_prompt:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+    
+  - from: "ckpt_loader:2"
+    to: "vae_decode:vae"
+  - from: "ckpt_loader:2"
+    to: "vae_encode:vae"
+
+dynamic_vae_chains:
+  vae_chain:
+    targets:
+      - "vae_decode:vae"
+      - "vae_encode:vae"
+
+dynamic_lora_chains:
+  lora_chain:
+    template: "LoraLoader"
+    start: "ckpt_loader" 
+    output_map:
+      "0": "model" 
+      "1": "clip"  
+    input_map:
+      "model": "model"
+      "clip": "clip"
+    end_input_map:
+      "model": ["model_sampler:model"]
+      "clip": ["pos_prompt:clip", "neg_prompt:clip"]
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    ksampler_node: "ksampler"
+    clip_source: "ckpt_loader:1"
+
+ui_map:
+  model_name: "ckpt_loader:ckpt_name"

core/pipelines/workflow_recipes/_partials/conditioning/newbie-image.yaml

@@ -0,0 +1,65 @@
+nodes:
+  unet_loader:
+    class_type: UNETLoader
+    title: "Load Diffusion Model"
+    params:
+      weight_dtype: "default"
+  vae_loader:
+    class_type: VAELoader
+    title: "Load VAE"
+  clip_loader:
+    class_type: DualCLIPLoader
+    title: "Load Dual CLIP"
+    params:
+      type: "newbie"
+      device: "default"
+  model_sampler:
+    class_type: ModelSamplingAuraFlow
+    title: "ModelSamplingAuraFlow"
+    params:
+      shift: 6
+
+connections:
+  - from: "unet_loader:0"
+    to: "model_sampler:model"
+  - from: "model_sampler:0"
+    to: "ksampler:model"
+  
+  - from: "clip_loader:0"
+    to: "pos_prompt:clip"
+  - from: "clip_loader:0"
+    to: "neg_prompt:clip"
+    
+  - from: "pos_prompt:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+  
+  - from: "vae_loader:0"
+    to: "vae_decode:vae"
+  - from: "vae_loader:0"
+    to: "vae_encode:vae"
+
+dynamic_newbie_lora_chains:
+  lora_chain:
+    template: "NewBieLoraLoader"
+    output_map:
+      "unet_loader:0": "model"
+      "clip_loader:0": "clip"
+    input_map:
+      "model": "model"
+      "clip": "clip"
+    end_input_map:
+      "model": ["model_sampler:model"]
+      "clip": ["pos_prompt:clip", "neg_prompt:clip"]
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    ksampler_node: "ksampler"
+    clip_source: "clip_loader:0"
+
+ui_map:
+  unet_name: "unet_loader:unet_name"
+  vae_name: "vae_loader:vae_name"
+  clip1_name: "clip_loader:clip_name1"
+  clip2_name: "clip_loader:clip_name2"

core/pipelines/workflow_recipes/_partials/conditioning/omnigen2.yaml

@@ -0,0 +1,59 @@
+nodes:
+  unet_loader:
+    class_type: UNETLoader
+    title: "Load Diffusion Model"
+    params:
+      weight_dtype: "default"
+  vae_loader:
+    class_type: VAELoader
+    title: "Load VAE"
+  clip_loader:
+    class_type: CLIPLoader
+    title: "Load CLIP"
+    params:
+      type: "omnigen2"
+      device: "default"
+
+connections:
+  - from: "unet_loader:0"
+    to: "ksampler:model"
+  - from: "clip_loader:0"
+    to: "pos_prompt:clip"
+  - from: "clip_loader:0"
+    to: "neg_prompt:clip"
+  - from: "pos_prompt:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+  - from: "vae_loader:0"
+    to: "vae_decode:vae"
+  - from: "vae_loader:0"
+    to: "vae_encode:vae"
+
+dynamic_lora_chains:
+  lora_chain:
+    template: "LoraLoader"
+    output_map:
+      "unet_loader:0": "model"
+      "clip_loader:0": "clip"
+    input_map:
+      "model": "model"
+      "clip": "clip"
+    end_input_map:
+      "model": ["ksampler:model"]
+      "clip": ["pos_prompt:clip", "neg_prompt:clip"]
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    ksampler_node: "ksampler"
+    clip_source: "clip_loader:0"
+
+dynamic_reference_latent_chains:
+  reference_latent_chain:
+    ksampler_node: "ksampler"
+    vae_node: "vae_loader"
+
+ui_map:
+  unet_name: "unet_loader:unet_name"
+  vae_name: "vae_loader:vae_name"
+  clip_name: "clip_loader:clip_name"

core/pipelines/workflow_recipes/_partials/conditioning/ovis-image.yaml

@@ -0,0 +1,50 @@
+nodes:
+  unet_loader:
+    class_type: UNETLoader
+    title: "Load Diffusion Model"
+    params:
+      weight_dtype: "default"
+  vae_loader:
+    class_type: VAELoader
+    title: "Load VAE"
+  clip_loader:
+    class_type: CLIPLoader
+    title: "Load CLIP"
+    params:
+      type: "ovis"
+      device: "default"
+  model_sampler:
+    class_type: ModelSamplingAuraFlow
+    params:
+      shift: 3.0
+
+connections:
+  - from: "unet_loader:0"
+    to: "model_sampler:model"
+  - from: "model_sampler:0"
+    to: "ksampler:model"
+  
+  - from: "clip_loader:0"
+    to: "pos_prompt:clip"
+  - from: "clip_loader:0"
+    to: "neg_prompt:clip"
+    
+  - from: "pos_prompt:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+  
+  - from: "vae_loader:0"
+    to: "vae_decode:vae"
+  - from: "vae_loader:0"
+    to: "vae_encode:vae"
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    ksampler_node: "ksampler"
+    clip_source: "clip_loader:0"
+
+ui_map:
+  unet_name: "unet_loader:unet_name"
+  vae_name: "vae_loader:vae_name"
+  clip_name: "clip_loader:clip_name"

core/pipelines/workflow_recipes/_partials/conditioning/qwen-image.yaml

@@ -0,0 +1,80 @@
+nodes:
+  unet_loader:
+    class_type: UNETLoader
+    title: "Load Qwen UNET"
+    params:
+      weight_dtype: "default"
+  vae_loader:
+    class_type: VAELoader
+    title: "Load Qwen VAE"
+  clip_loader:
+    class_type: CLIPLoader
+    title: "Load Qwen CLIP"
+    params:
+      type: "qwen_image"
+      device: "default"
+  
+  lora_loader:
+    class_type: LoraLoaderModelOnly
+    title: "Load Qwen Lightning LoRA"
+    params:
+      strength_model: 1.0
+  model_sampler:
+    class_type: ModelSamplingAuraFlow
+    title: "ModelSamplingAuraFlow"
+    params:
+      shift: 3.1
+
+connections:
+  - from: "unet_loader:0"
+    to: "lora_loader:model"
+  - from: "lora_loader:0"
+    to: "model_sampler:model"
+
+  - from: "model_sampler:0"
+    to: "ksampler:model"
+    
+  - from: "clip_loader:0"
+    to: "pos_prompt:clip"
+  - from: "clip_loader:0"
+    to: "neg_prompt:clip"
+    
+  - from: "vae_loader:0"
+    to: "vae_decode:vae"
+  - from: "vae_loader:0"
+    to: "vae_encode:vae"
+
+  - from: "pos_prompt:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+
+dynamic_lora_chains:
+  lora_chain:
+    template: "LoraLoader"
+    output_map:
+      "lora_loader:0": "model"
+      "clip_loader:0": "clip"
+    input_map:
+      "model": "model"
+      "clip": "clip"
+    end_input_map:
+      "model": ["model_sampler:model"]
+      "clip": ["pos_prompt:clip", "neg_prompt:clip"]
+
+dynamic_controlnet_chains:
+  controlnet_chain:
+    template: "ControlNetApplyAdvanced"
+    ksampler_node: "ksampler"
+    vae_source: "vae_loader:0"
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    ksampler_node: "ksampler"
+    clip_source: "clip_loader:0"
+
+ui_map:
+  unet_name: "unet_loader:unet_name"
+  vae_name: "vae_loader:vae_name"
+  clip_name: "clip_loader:clip_name"
+  lora_name: "lora_loader:lora_name"

core/pipelines/workflow_recipes/_partials/conditioning/sd15.yaml

@@ -0,0 +1,69 @@
+nodes:
+  ckpt_loader:
+    class_type: CheckpointLoaderSimple
+    title: "Load Checkpoint"
+  clip_set_last_layer:
+    class_type: CLIPSetLastLayer
+    title: "CLIP Set Last Layer"
+
+connections:
+  - from: "ckpt_loader:0"
+    to: "ksampler:model"
+  - from: "ckpt_loader:1"
+    to: "clip_set_last_layer:clip"
+  - from: "clip_set_last_layer:0"
+    to: "pos_prompt:clip"
+  - from: "clip_set_last_layer:0"
+    to: "neg_prompt:clip"
+  - from: "pos_prompt:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+  - from: "ckpt_loader:2"
+    to: "vae_decode:vae"
+  - from: "ckpt_loader:2"
+    to: "vae_encode:vae"
+
+dynamic_vae_chains:
+  vae_chain:
+    targets:
+      - "vae_decode:vae"
+      - "vae_encode:vae"
+
+dynamic_lora_chains:
+  lora_chain:
+    template: "LoraLoader"
+    start: "clip_set_last_layer"
+    output_map:
+      "ckpt_loader:0": "model" 
+      "0": "clip"             
+    input_map:
+      "model": "model"
+      "clip": "clip"
+    end_input_map:
+      "model": ["ksampler:model"]
+      "clip": ["pos_prompt:clip", "neg_prompt:clip"]
+
+dynamic_controlnet_chains:
+  controlnet_chain:
+    template: "ControlNetApplyAdvanced"
+    ksampler_node: "ksampler"
+    vae_source: "ckpt_loader:2"
+
+dynamic_ipadapter_chains:
+  ipadapter_chain:
+    end: "ksampler"
+    final_preset: "{{ ipadapter_final_preset }}"
+    final_weight: "{{ ipadapter_final_weight }}"
+    final_embeds_scaling: "{{ ipadapter_embeds_scaling }}"
+    final_loader_type: "{{ ipadapter_final_loader_type }}"
+    final_lora_strength: "{{ ipadapter_final_lora_strength }}"
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    ksampler_node: "ksampler"
+    clip_source: "clip_set_last_layer:0"
+
+ui_map:
+  model_name: "ckpt_loader:ckpt_name"
+  clip_skip: "clip_set_last_layer:stop_at_clip_layer"

core/pipelines/workflow_recipes/_partials/conditioning/sd35.yaml

@@ -0,0 +1,58 @@
+nodes:
+  ckpt_loader:
+    class_type: CheckpointLoaderSimple
+    title: "Load Checkpoint"
+
+connections:
+  - from: "ckpt_loader:0"
+    to: "ksampler:model"
+  - from: "ckpt_loader:1"
+    to: "pos_prompt:clip"
+  - from: "ckpt_loader:1"
+    to: "neg_prompt:clip"
+  - from: "pos_prompt:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+  - from: "ckpt_loader:2"
+    to: "vae_decode:vae"
+  - from: "ckpt_loader:2"
+    to: "vae_encode:vae"
+
+dynamic_vae_chains:
+  vae_chain:
+    targets:
+      - "vae_decode:vae"
+      - "vae_encode:vae"
+
+dynamic_lora_chains:
+  lora_chain:
+    template: "LoraLoader"
+    start: "ckpt_loader" 
+    output_map:
+      "0": "model" 
+      "1": "clip"  
+    input_map:
+      "model": "model"
+      "clip": "clip"
+    end_input_map:
+      "model": ["ksampler:model"]
+      "clip": ["pos_prompt:clip", "neg_prompt:clip"]
+
+dynamic_controlnet_chains:
+  controlnet_chain:
+    template: "ControlNetApplyAdvanced"
+    ksampler_node: "ksampler"
+    vae_source: "ckpt_loader:2"
+
+dynamic_sd3_ipadapter_chains:
+  sd3_ipadapter_chain:
+    ksampler_node: "ksampler"
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    ksampler_node: "ksampler"
+    clip_source: "ckpt_loader:1"
+
+ui_map:
+  model_name: "ckpt_loader:ckpt_name"

core/pipelines/workflow_recipes/_partials/conditioning/sdxl.yaml

@@ -0,0 +1,63 @@
+nodes:
+  ckpt_loader:
+    class_type: CheckpointLoaderSimple
+    title: "Load Checkpoint"
+
+connections:
+  - from: "ckpt_loader:0"
+    to: "ksampler:model"
+  - from: "ckpt_loader:1"
+    to: "pos_prompt:clip"
+  - from: "ckpt_loader:1"
+    to: "neg_prompt:clip"
+  - from: "pos_prompt:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+  - from: "ckpt_loader:2"
+    to: "vae_decode:vae"
+  - from: "ckpt_loader:2"
+    to: "vae_encode:vae"
+
+dynamic_vae_chains:
+  vae_chain:
+    targets:
+      - "vae_decode:vae"
+      - "vae_encode:vae"
+
+dynamic_lora_chains:
+  lora_chain:
+    template: "LoraLoader"
+    start: "ckpt_loader" 
+    output_map:
+      "0": "model" 
+      "1": "clip"  
+    input_map:
+      "model": "model"
+      "clip": "clip"
+    end_input_map:
+      "model": ["ksampler:model"]
+      "clip": ["pos_prompt:clip", "neg_prompt:clip"]
+
+dynamic_controlnet_chains:
+  controlnet_chain:
+    template: "ControlNetApplyAdvanced"
+    ksampler_node: "ksampler"
+    vae_source: "ckpt_loader:2"
+
+dynamic_ipadapter_chains:
+  ipadapter_chain:
+    end: "ksampler"
+    final_preset: "{{ ipadapter_final_preset }}"
+    final_weight: "{{ ipadapter_final_weight }}"
+    final_embeds_scaling: "{{ ipadapter_embeds_scaling }}"
+    final_loader_type: "{{ ipadapter_final_loader_type }}"
+    final_lora_strength: "{{ ipadapter_final_lora_strength }}"
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    ksampler_node: "ksampler"
+    clip_source: "ckpt_loader:1"
+
+ui_map:
+  model_name: "ckpt_loader:ckpt_name"

core/pipelines/workflow_recipes/_partials/conditioning/z-image.yaml

@@ -0,0 +1,65 @@
+nodes:
+  unet_loader:
+    class_type: UNETLoader
+    title: "Load Diffusion Model"
+    params:
+      weight_dtype: "default"
+  vae_loader:
+    class_type: VAELoader
+    title: "Load VAE"
+  clip_loader:
+    class_type: CLIPLoader
+    title: "Load CLIP"
+    params:
+      type: "lumina2"
+      device: "default"
+  model_sampler:
+    class_type: ModelSamplingAuraFlow
+    params:
+      shift: 3.0
+
+connections:
+  - from: "unet_loader:0"
+    to: "model_sampler:model"
+  - from: "model_sampler:0"
+    to: "ksampler:model"
+  
+  - from: "clip_loader:0"
+    to: "pos_prompt:clip"
+  - from: "clip_loader:0"
+    to: "neg_prompt:clip"
+    
+  - from: "pos_prompt:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+  
+  - from: "vae_loader:0"
+    to: "vae_decode:vae"
+  - from: "vae_loader:0"
+    to: "vae_encode:vae"
+
+dynamic_lora_chains:
+  lora_chain:
+    template: "LoraLoader"
+    output_map:
+      "unet_loader:0": "model"
+      "clip_loader:0": "clip"
+    input_map:
+      "model": "model"
+      "clip": "clip"
+    end_input_map:
+      "model": ["model_sampler:model"]
+      "clip": ["pos_prompt:clip", "neg_prompt:clip"]
+
+dynamic_diffsynth_controlnet_chains:
+  diffsynth_controlnet_chain:
+    template: "QwenImageDiffsynthControlnet"
+    model_sampler_node: "model_sampler"
+    ksampler_node: "ksampler"
+    vae_source: "vae_loader:0"
+
+ui_map:
+  unet_name: "unet_loader:unet_name"
+  vae_name: "vae_loader:vae_name"
+  clip_name: "clip_loader:clip_name"

core/pipelines/workflow_recipes/_partials/input/hires_fix.yaml

@@ -1,15 +1,16 @@
 nodes:
   input_image_loader:
     class_type: LoadImage
-    
+    title: "Load Input Image"
   vae_encode:
     class_type: VAEEncode
-    
+    title: "VAE Encode (Hires Pre-step)"
   latent_upscaler:
     class_type: LatentUpscaleBy
-
+    title: "Upscale Latent By"
   latent_source:
     class_type: RepeatLatentBatch
+    title: "Repeat Latent Batch for Hires"
 
 connections:
   - from: "input_image_loader:0"

core/pipelines/workflow_recipes/_partials/input/img2img.yaml

@@ -1,12 +1,13 @@
 nodes:
   input_image_loader:
     class_type: LoadImage
-  
+    title: "Load Input Image"
   vae_encode:
     class_type: VAEEncode
-
+    title: "VAE Encode (Img2Img)"
   latent_source:
     class_type: RepeatLatentBatch
+    title: "Repeat Latent Batch"
 
 connections:
   - from: "input_image_loader:0"

core/pipelines/workflow_recipes/_partials/input/inpaint.yaml

@@ -2,24 +2,22 @@ nodes:
   inpaint_loader:
     class_type: LoadImage
     title: "Load Inpaint Image+Mask"
-      
   vae_encode:
     class_type: VAEEncodeForInpaint
-    params:
-      grow_mask_by: 6
-      
+    title: "VAE Encode (for Inpainting)"
   latent_source:
     class_type: RepeatLatentBatch
-      
+    title: "Repeat Latent Batch"
+
 connections:
   - from: "inpaint_loader:0"
     to: "vae_encode:pixels"
   - from: "inpaint_loader:1"
     to: "vae_encode:mask"
-    
   - from: "vae_encode:0"
     to: "latent_source:samples"
 
 ui_map:
-  inpaint_image: "inpaint_loader:image"
-  batch_size: "latent_source:amount"
+  input_image: "inpaint_loader:image"
+  batch_size: "latent_source:amount"
+  grow_mask_by: "vae_encode:grow_mask_by"

core/pipelines/workflow_recipes/_partials/input/outpaint.yaml

@@ -1,38 +1,41 @@
 nodes:
   input_image_loader:
     class_type: LoadImage
-    
+    title: "Load Image for Outpaint"
+  scale_image:
+    class_type: ImageScaleToTotalPixels
+    title: "Scale Image to Total Pixels"
+    params:
+      upscale_method: "nearest-exact"
   pad_image:
     class_type: ImagePadForOutpaint
-    params:
-      feathering: 10
-      
+    title: "Pad Image for Outpainting"
   vae_encode:
     class_type: VAEEncodeForInpaint
-    params:
-      grow_mask_by: 6
-
+    title: "VAE Encode (for Inpainting)"
   latent_source:
     class_type: RepeatLatentBatch
+    title: "Repeat Latent Batch"
 
 connections:
   - from: "input_image_loader:0"
+    to: "scale_image:image"
+  - from: "scale_image:0"
     to: "pad_image:image"
-    
   - from: "pad_image:0"
     to: "vae_encode:pixels"
   - from: "pad_image:1"
     to: "vae_encode:mask"
-
   - from: "vae_encode:0"
     to: "latent_source:samples"
 
 ui_map:
   input_image: "input_image_loader:image"
-  
+  megapixels: "scale_image:megapixels"
   left: "pad_image:left"
   top: "pad_image:top"
   right: "pad_image:right"
   bottom: "pad_image:bottom"
-  
+  feathering: "pad_image:feathering"
+  grow_mask_by: "vae_encode:grow_mask_by"
   batch_size: "latent_source:amount"

core/pipelines/workflow_recipes/_partials/input/txt2img.yaml

@@ -1,8 +1,2 @@
-nodes:
-  latent_source:
-    class_type: EmptyFlux2LatentImage
-
-ui_map:
-  width: "latent_source:width"
-  height: "latent_source:height"
-  batch_size: "latent_source:batch_size"
+imports:
+  - "txt2img_{{ latent_type }}.yaml"

core/pipelines/workflow_recipes/_partials/input/txt2img_chroma_radiance_latent.yaml

@@ -0,0 +1,11 @@
+nodes:
+  latent_source:
+    class_type: "EmptyChromaRadianceLatentImage"
+    title: "EmptyChromaRadianceLatentImage"
+
+connections: []
+
+ui_map:
+  width: "latent_source:width"
+  height: "latent_source:height"
+  batch_size: "latent_source:batch_size"

core/pipelines/workflow_recipes/_partials/input/txt2img_flux2_latent.yaml

@@ -0,0 +1,11 @@
+nodes:
+  latent_source:
+    class_type: "EmptyFlux2LatentImage"
+    title: "Empty Flux 2 Latent"
+
+connections: []
+
+ui_map:
+  width: "latent_source:width"
+  height: "latent_source:height"
+  batch_size: "latent_source:batch_size"

core/pipelines/workflow_recipes/_partials/input/txt2img_hunyuan_latent.yaml

@@ -0,0 +1,11 @@
+nodes:
+  latent_source:
+    class_type: "EmptyHunyuanImageLatent"
+    title: "EmptyHunyuanImageLatent"
+
+connections: []
+
+ui_map:
+  width: "latent_source:width"
+  height: "latent_source:height"
+  batch_size: "latent_source:batch_size"

core/pipelines/workflow_recipes/_partials/input/txt2img_latent.yaml

@@ -0,0 +1,11 @@
+nodes:
+  latent_source:
+    class_type: "{{ latent_generator_template }}"
+    title: "Empty Latent Image"
+
+connections: []
+
+ui_map:
+  width: "latent_source:width"
+  height: "latent_source:height"
+  batch_size: "latent_source:batch_size"

core/pipelines/workflow_recipes/_partials/input/txt2img_sd3_latent.yaml

@@ -0,0 +1,11 @@
+nodes:
+  latent_source:
+    class_type: "EmptySD3LatentImage"
+    title: "EmptySD3LatentImage"
+
+connections: []
+
+ui_map:
+  width: "latent_source:width"
+  height: "latent_source:height"
+  batch_size: "latent_source:batch_size"

core/pipelines/workflow_recipes/sd_unified_recipe.yaml

@@ -1,7 +1,7 @@
 imports:
-  - "_partials/_base_sampler.yaml"
+  - "_partials/_base_sampler_sd.yaml"
   - "_partials/input/{{ task_type }}.yaml"
-  - "_partials/conditioning/flux2.yaml"
+  - "_partials/conditioning/{{ model_type }}.yaml"
 
 connections:
   - from: "latent_source:0"

core/settings.py

@@ -10,16 +10,37 @@ MODEL_PATCHES_DIR = "models/model_patches"
 DIFFUSION_MODELS_DIR = "models/diffusion_models"
 VAE_DIR = "models/vae"
 TEXT_ENCODERS_DIR = "models/text_encoders"
+STYLE_MODELS_DIR = "models/style_models"
+CLIP_VISION_DIR = "models/clip_vision"
+IPADAPTER_DIR = "models/ipadapter"
+IPADAPTER_FLUX_DIR = "models/ipadapter-flux"
 INPUT_DIR = "input"
 OUTPUT_DIR = "output"
 
+CATEGORY_TO_DIR_MAP = {
+    "diffusion_models": DIFFUSION_MODELS_DIR,
+    "text_encoders": TEXT_ENCODERS_DIR,
+    "vae": VAE_DIR,
+    "checkpoints": CHECKPOINT_DIR,
+    "loras": LORA_DIR,
+    "controlnet": CONTROLNET_DIR,
+    "model_patches": MODEL_PATCHES_DIR,
+    "embeddings": EMBEDDING_DIR,
+    "style_models": STYLE_MODELS_DIR,
+    "clip_vision": CLIP_VISION_DIR,
+    "ipadapter": IPADAPTER_DIR,
+    "ipadapter-flux": IPADAPTER_FLUX_DIR
+}
+
 _PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 _MODEL_LIST_PATH = os.path.join(_PROJECT_ROOT, 'yaml', 'model_list.yaml')
 _FILE_LIST_PATH = os.path.join(_PROJECT_ROOT, 'yaml', 'file_list.yaml')
+_IPADAPTER_LIST_PATH = os.path.join(_PROJECT_ROOT, 'yaml', 'ipadapter.yaml')
 _CONSTANTS_PATH = os.path.join(_PROJECT_ROOT, 'yaml', 'constants.yaml')
+_MODEL_ARCHITECTURES_PATH = os.path.join(_PROJECT_ROOT, 'yaml', 'model_architectures.yaml')
+_IMAGE_GEN_FEATURES_PATH = os.path.join(_PROJECT_ROOT, 'yaml', 'image_gen_features.yaml')
 _MODEL_DEFAULTS_PATH = os.path.join(_PROJECT_ROOT, 'yaml', 'model_defaults.yaml')
 
-
 def load_constants_from_yaml(filepath=_CONSTANTS_PATH):
     if not os.path.exists(filepath):
         print(f"Warning: Constants file not found at {filepath}. Using fallback values.")
@@ -27,6 +48,27 @@ def load_constants_from_yaml(filepath=_CONSTANTS_PATH):
     with open(filepath, 'r', encoding='utf-8') as f:
         return yaml.safe_load(f)
 
+def load_architectures_config(filepath=_MODEL_ARCHITECTURES_PATH):
+    if not os.path.exists(filepath):
+        print(f"Warning: Architectures file not found at {filepath}.")
+        return {}
+    with open(filepath, 'r', encoding='utf-8') as f:
+        return yaml.safe_load(f)
+
+def load_features_config(filepath=_IMAGE_GEN_FEATURES_PATH):
+    if not os.path.exists(filepath):
+        print(f"Warning: Features file not found at {filepath}.")
+        return {}
+    with open(filepath, 'r', encoding='utf-8') as f:
+        return yaml.safe_load(f)
+
+def load_model_defaults(filepath=_MODEL_DEFAULTS_PATH):
+    if not os.path.exists(filepath):
+        print(f"Warning: Model defaults file not found at {filepath}.")
+        return {}
+    with open(filepath, 'r', encoding='utf-8') as f:
+        return yaml.safe_load(f)
+
 def load_file_download_map(filepath=_FILE_LIST_PATH):
     if not os.path.exists(filepath):
         raise FileNotFoundError(f"The file list (for downloads) was not found at: {filepath}")
@@ -59,50 +101,86 @@ def load_models_from_yaml(model_list_filepath=_MODEL_LIST_PATH, download_map=Non
     }
     category_map_names = {
         "Checkpoint": "MODEL_MAP_CHECKPOINT",
+        "Checkpoints": "MODEL_MAP_CHECKPOINT"
     }
     
-    for category, models in model_data.items():
+    for category, architectures in model_data.items():
         if category in category_map_names:
             map_name = category_map_names[category]
-            if not isinstance(models, list): continue
-            for model in models:
-                display_name = model['display_name']
-                components = model.get('components', {})
-
-                model_tuple = (
-                    None,
-                    components,
-                    "SDXL",
-                    None
-                )
-                model_maps[map_name][display_name] = model_tuple
-                model_maps["ALL_MODEL_MAP"][display_name] = model_tuple
+            if not isinstance(architectures, dict): continue
+            
+            for arch, arch_data in architectures.items():
+                if not isinstance(arch_data, dict): continue
+                
+                latent_type = arch_data.get('latent_type', 'latent')
+                models = arch_data.get('models', [])
+                if not isinstance(models, list): continue
+                
+                for model in models:
+                    display_name = model['display_name']
+                    path_or_components = model.get('path') or model.get('components')
+                    mod_category = model.get('category', None)
+                    
+                    repo_id = ''
+                    if isinstance(path_or_components, str):
+                        download_info = download_map.get(path_or_components, {})
+                        repo_id = download_info.get('repo_id', '')
+                    
+                    model_tuple = (
+                        repo_id,
+                        path_or_components,
+                        arch,
+                        latent_type,
+                        mod_category
+                    )
+                    model_maps[map_name][display_name] = model_tuple
+                    model_maps["ALL_MODEL_MAP"][display_name] = model_tuple
     
     return model_maps
 
-def load_model_defaults(filepath=_MODEL_DEFAULTS_PATH):
-    if not os.path.exists(filepath):
-        print(f"Warning: Model defaults file not found at {filepath}. Using empty defaults.")
-        return {}
-    with open(filepath, 'r', encoding='utf-8') as f:
-        return yaml.safe_load(f)
-
 try:
     ALL_FILE_DOWNLOAD_MAP = load_file_download_map()
     loaded_maps = load_models_from_yaml(download_map=ALL_FILE_DOWNLOAD_MAP)
     MODEL_MAP_CHECKPOINT = loaded_maps["MODEL_MAP_CHECKPOINT"]
     ALL_MODEL_MAP = loaded_maps["ALL_MODEL_MAP"]
     
+    category_to_model_type = {
+        "diffusion_models": "UNET",
+        "text_encoders": "TEXT_ENCODER",
+        "vae": "VAE",
+        "checkpoints": "SDXL",
+        "loras": "LORA",
+        "controlnet": "CONTROLNET",
+        "model_patches": "MODEL_PATCH",
+        "style_models": "STYLE",
+        "clip_vision": "CLIP_VISION",
+        "ipadapter": "IPADAPTER",
+        "ipadapter-flux": "IPADAPTER_FLUX"
+    }
+    for filename, file_info in ALL_FILE_DOWNLOAD_MAP.items():
+        if filename not in ALL_MODEL_MAP:
+            category = file_info.get('category')
+            model_type = category_to_model_type.get(category, 'UNKNOWN')
+            repo_id = file_info.get('repo_id', '')
+            ALL_MODEL_MAP[filename] = (repo_id, filename, model_type, None, None)
+
     MODEL_TYPE_MAP = {k: v[2] for k, v in ALL_MODEL_MAP.items()}
-    
-    ALL_MODEL_DEFAULTS = load_model_defaults()
+
+    ARCH_CATEGORIES_MAP = {}
+    for display_name, info in MODEL_MAP_CHECKPOINT.items():
+        arch = info[2]
+        cat = info[4] if len(info) > 4 else None
+        if arch not in ARCH_CATEGORIES_MAP:
+            ARCH_CATEGORIES_MAP[arch] = []
+        if cat and cat not in ARCH_CATEGORIES_MAP[arch]:
+            ARCH_CATEGORIES_MAP[arch].append(cat)
 
 except Exception as e:
     print(f"FATAL: Could not load model configuration from YAML. Error: {e}")
     ALL_FILE_DOWNLOAD_MAP = {}
     MODEL_MAP_CHECKPOINT, ALL_MODEL_MAP = {}, {}
     MODEL_TYPE_MAP = {}
-    ALL_MODEL_DEFAULTS = {}
+    ARCH_CATEGORIES_MAP = {}
 
 
 try:
@@ -111,15 +189,17 @@ try:
     MAX_EMBEDDINGS = _constants.get('MAX_EMBEDDINGS', 5)
     MAX_CONDITIONINGS = _constants.get('MAX_CONDITIONINGS', 10)
     MAX_CONTROLNETS = _constants.get('MAX_CONTROLNETS', 5)
-    MAX_REFERENCE_LATENTS = _constants.get('MAX_REFERENCE_LATENTS', 10)
+    MAX_IPADAPTERS = _constants.get('MAX_IPADAPTERS', 5)
     LORA_SOURCE_CHOICES = _constants.get('LORA_SOURCE_CHOICES', ["Civitai", "File"])
     RESOLUTION_MAP = _constants.get('RESOLUTION_MAP', {})
+    ARCHITECTURES_CONFIG = load_architectures_config()
+    FEATURES_CONFIG = load_features_config()
+    MODEL_DEFAULTS_CONFIG = load_model_defaults()
 except Exception as e:
     print(f"FATAL: Could not load constants from YAML. Error: {e}")
-    MAX_LORAS, MAX_EMBEDDINGS, MAX_CONDITIONINGS, MAX_CONTROLNETS = 5, 5, 10, 5
-    MAX_REFERENCE_LATENTS = 10
+    MAX_LORAS, MAX_EMBEDDINGS, MAX_CONDITIONINGS, MAX_CONTROLNETS, MAX_IPADAPTERS = 5, 5, 10, 5, 5
     LORA_SOURCE_CHOICES = ["Civitai", "File"]
     RESOLUTION_MAP = {}
-
-
-DEFAULT_NEGATIVE_PROMPT = ""
+    ARCHITECTURES_CONFIG = {}
+    FEATURES_CONFIG = {}
+    MODEL_DEFAULTS_CONFIG = {}

requirements.txt

@@ -1,6 +1,6 @@
-comfyui-frontend-package==1.42.14
-comfyui-workflow-templates==0.9.59
-comfyui-embedded-docs==0.4.3
+comfyui-frontend-package==1.42.15
+comfyui-workflow-templates==0.9.66
+comfyui-embedded-docs==0.4.4
 torch==2.10.0
 torchsde
 torchvision==0.25.0
@@ -19,11 +19,11 @@ scipy
 tqdm
 psutil
 alembic
-SQLAlchemy>=2.0
+SQLAlchemy>=2.0.0
 filelock
 av>=14.2.0
 comfy-kitchen>=0.2.8
-comfy-aimdo>=0.2.12
+comfy-aimdo==0.3.0
 requests
 simpleeval>=1.0.0
 blake3
@@ -58,4 +58,5 @@ svglib
 trimesh[easy]
 yacs
 yapf
-onnxruntime-gpu
+onnxruntime-gpu
+diffusers