Add a way to set patches that modify the attn2 output.

Change the transformer patches function format to be more future proof.
This commit is contained in:
comfyanonymous
2023-06-18 22:58:22 -04:00
parent cd930d4e7f
commit 8883cb0f67
4 changed files with 19 additions and 7 deletions

View File

@@ -260,7 +260,8 @@ class Gligen(nn.Module):
return r
return func_lowvram
else:
def func(key, x):
def func(x, extra_options):
key = extra_options["transformer_index"]
module = self.module_list[key]
return module(x, objs)
return func