Browse Source

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

pull/1619/head
comfyanonymous 1 year ago
parent
commit
fff491b032
  1. 2
      comfy/ldm/modules/attention.py

2
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:

Loading…
Cancel
Save