Upload folder using huggingface_hub

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}).")

core/pipelines/sd_image_pipeline.py

@@ -490,6 +490,7 @@ class SdImagePipeline(BasePipeline):
             "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):
@@ -512,20 +513,6 @@ class SdImagePipeline(BasePipeline):
         
         workflow = assembler.assemble(workflow_inputs)
 
-        if ui_inputs.get("vae_name") and workflow_model_type not in ['flux1', 'hidream', 'lumina', 'omnigen2', 'chroma1-radiance', 'chroma1', 'hunyuanimage', 'ovis-image', 'longcat-image']:
-            print("--- [Workflow Patch] VAE override provided. Adding VAELoader and rewiring connections. ---")
-            vae_loader_id = assembler._get_unique_id()
-            vae_loader_node = assembler._get_node_template("VAELoader")
-            vae_loader_node['inputs']['vae_name'] = ui_inputs["vae_name"]
-            workflow[vae_loader_id] = vae_loader_node
-
-            vae_decode_id = assembler.node_map.get("vae_decode")
-            if vae_decode_id and vae_decode_id in workflow:
-                workflow[vae_decode_id]['inputs']['vae'] = [vae_loader_id, 0]
-            vae_encode_id = assembler.node_map.get("vae_encode")
-            if vae_encode_id and vae_encode_id in workflow:
-                workflow[vae_encode_id]['inputs']['vae'] = [vae_loader_id, 0]
-        
         progress(1.0, desc="All models ready. Requesting GPU for generation...")
 
         try:

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

@@ -28,6 +28,12 @@ connections:
   - 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"

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

@@ -24,6 +24,12 @@ connections:
   - 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"

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

@@ -19,6 +19,12 @@ connections:
   - 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"

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

@@ -19,6 +19,12 @@ connections:
   - 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"

yaml/injectors.yaml

@@ -1,4 +1,6 @@
 injector_definitions:
+  dynamic_vae_chains:
+    module: "chain_injectors.vae_injector"
   dynamic_lora_chains: 
     module: "chain_injectors.lora_injector"
   dynamic_newbie_lora_chains:
@@ -21,6 +23,7 @@ injector_definitions:
     module: "chain_injectors.reference_latent_injector"
   
 injector_order:
+  - dynamic_vae_chains
   - dynamic_lora_chains
   - dynamic_newbie_lora_chains
   - dynamic_diffsynth_controlnet_chains