From 43d4935a1da0b78dac101a28cc98de0b7d556729 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Fri, 15 Sep 2023 22:21:14 -0400 Subject: [PATCH] Add cond_or_uncond array to transformer_options so hooks can check what is cond and what is uncond. --- comfy/samplers.py | 1 + 1 file changed, 1 insertion(+) 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: