Skip to content

GitLab

"git@developer.sourcefind.cn:gaoqiong/rapidasr.git" did not exist on "c1a342ac54abed6249347e31d27dd507d7859291"
Commit fff491b0 authored by comfyanonymous's avatar comfyanonymous
Browse files

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

parent 1d6dd831
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
comfy/ldm/modules/attention.py
View file @ fff491b0
...@@ -538,6 +538,8 @@ class BasicTransformerBlock(nn.Module): ...@@ -538,6 +538,8 @@ class BasicTransformerBlock(nn.Module):
if "block" in transformer_options: if "block" in transformer_options:
block = transformer_options["block"] block = transformer_options["block"]
extra_options["block"] = 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: if "patches" in transformer_options:
transformer_patches = transformer_options["patches"] transformer_patches = transformer_options["patches"]
else: else:
......
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