diff --git a/modules_forge/forge_loader.py b/modules_forge/forge_loader.py
index 33cb7fde..0c68c2cf 100644
--- a/modules_forge/forge_loader.py
+++ b/modules_forge/forge_loader.py
@@ -241,9 +241,19 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None):
     sd_model.decode_first_stage = patched_decode_first_stage
     sd_model.encode_first_stage = patched_encode_first_stage
 
+    sd_model.current_control_signals = {
+        'input': [],
+        'middle': [],
+        'output': []
+    }
+
     original_forward = sd_model.model.diffusion_model.forward
 
     def forge_unet_forward(*args, **kwargs):
+        kwargs.update(dict(
+            control=sd_model.current_control_signals,
+            transformer_options=sd_model.forge_objects.unet.model_options.get('transformer_options', {})
+        ))
         return original_forward(*args, **kwargs)
 
     sd_model.model.diffusion_model.forward = forge_unet_forward