ip adapter masking

This commit is contained in:
Vladimir Mandic
2024-04-18 12:43:51 -04:00
parent 77909a103a
commit f2610c3936
12 changed files with 84 additions and 31 deletions
+7
View File
@@ -282,6 +282,12 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
args[k] = v
else:
debug(f'Diffusers unknown task args: {k}={v}')
cross_attention_args = getattr(p, 'cross_attention_kwargs', {})
debug(f'Diffusers cross-attention args: {cross_attention_args}')
for k, v in cross_attention_args.items():
if args.get('cross_attention_kwargs', None) is None:
args['cross_attention_kwargs'] = {}
args['cross_attention_kwargs'][k] = v
# handle implicit controlnet
if 'control_image' in possible and 'control_image' not in args and 'image' in args:
@@ -292,6 +298,7 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
# debug info
clean = args.copy()
clean.pop('cross_attention_kwargs', None)
clean.pop('callback', None)
clean.pop('callback_steps', None)
clean.pop('callback_on_step_end', None)