Upload folder using huggingface_hub

chain_injectors/controlnet_injector.py

@@ -13,6 +13,16 @@ def inject(assembler, chain_definition, chain_items):
        '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']
@@ -37,6 +47,7 @@ def inject(assembler, chain_definition, chain_items):
         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
 

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

@@ -36,6 +36,7 @@ dynamic_controlnet_chains:
   controlnet_chain:
     template: "ControlNetApplyAdvanced"
     ksampler_node: "ksampler"
+    vae_source: "ckpt_loader:2"
 
 dynamic_ipadapter_chains:
   ipadapter_chain: