Add cond_or_uncond array to transformer_options so hooks can check what is

cond and what is uncond.
This commit is contained in:
comfyanonymous 2023-09-15 22:21:14 -04:00
parent 415abb275f
commit 43d4935a1d

View File

@ -255,6 +255,7 @@ def sampling_function(model_function, x, timestep, uncond, cond, cond_scale, con
else: else:
transformer_options["patches"] = patches transformer_options["patches"] = patches
transformer_options["cond_or_uncond"] = cond_or_uncond[:]
c['transformer_options'] = transformer_options c['transformer_options'] = transformer_options
if 'model_function_wrapper' in model_options: if 'model_function_wrapper' in model_options: