Model patches can now know which batch is positive and negative.

1 year ago · fff491b032
parent 1d6dd83184
commit fff491b032
1 changed files with 2 additions and 0 deletions
--- a/comfy/ldm/modules/attention.py
+++ b/comfy/ldm/modules/attention.py
@ -538,6 +538,8 @@ class BasicTransformerBlock(nn.Module):
        if "block" in transformer_options:
            block = transformer_options["block"]
            extra_options["block"] = block
+        if "cond_or_uncond" in transformer_options:
+            extra_options["cond_or_uncond"] = transformer_options["cond_or_uncond"]
        if "patches" in transformer_options:
            transformer_patches = transformer_options["patches"]
        else: