Commit 43d4935a authored by comfyanonymous's avatar comfyanonymous
Browse files

Add cond_or_uncond array to transformer_options so hooks can check what is

cond and what is uncond.
parent 415abb27
......@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment