diff --git a/comfy/samplers.py b/comfy/samplers.py
index 7f1987167..57673a029 100644
--- a/comfy/samplers.py
+++ b/comfy/samplers.py
@@ -255,6 +255,7 @@ def sampling_function(model_function, x, timestep, uncond, cond, cond_scale, con
                     else:
                         transformer_options["patches"] = patches
 
+                transformer_options["cond_or_uncond"] = cond_or_uncond[:]
                 c['transformer_options'] = transformer_options
 
                 if 'model_function_wrapper' in model_options: