Skip to content

GitLab

"git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "d623bf9b31f82b80cc093486bd701969bb860993"
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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
comfy/samplers.py
View file @ 43d4935a
...@@ -255,6 +255,7 @@ def sampling_function(model_function, x, timestep, uncond, cond, cond_scale, con ...@@ -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:
......
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