Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-08-29 16:41:01 +02:00
Code Issues Packages Projects Releases Wiki Activity

add task_args

Browse Source
This commit is contained in:
Vladimir Mandic
2023-11-24 08:30:30 -05:00
parent 29461cb6f2
commit 9edfc987e0
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_diffusers.py
+4 -2
View File
@@ -311,8 +311,8 @@ def process_diffusers(p: StableDiffusionProcessing, seeds, prompts, negative_pro
args[arg] = task_kwargs[arg]
else:
pass
# shared.log.debug(f'Diffuser not supported: pipeline={pipeline.__class__.__name__} task={sd_models.get_diffusers_task(model)} arg={arg}')
# shared.log.debug(f'Diffuser pipeline: {model.__class__.__name__} possible={possible}')
for k, v in getattr(p, 'task_args', {}).items():
args[k] = v
hypertile_set(p, hr=len(getattr(p, 'init_images', [])))
clean = args.copy()
clean.pop('callback', None)
@@ -327,6 +327,8 @@ def process_diffusers(p: StableDiffusionProcessing, seeds, prompts, negative_pro
clean['mask_image'] = type(clean['mask_image'])
if 'masked_image_latents' in clean:
clean['masked_image_latents'] = type(clean['masked_image_latents'])
if 'ip_adapter_image' in clean:
clean['ip_adapter_image'] = type(clean['ip_adapter_image'])
if 'prompt' in clean:
clean['prompt'] = len(clean['prompt'])
if 'negative_prompt' in clean: