Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

sampler flow shift options and fix img2img

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-12-27 16:33:19 -05:00
parent 19aac2d3c5
commit 7b7f121a96
8 changed files with 32 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_args.py
+4 -1
View File
@@ -22,7 +22,7 @@ def task_specific_kwargs(p, model):
if len(getattr(p, 'init_images', [])) > 0:
if isinstance(p.init_images[0], str):
p.init_images = [helpers.decode_base64_to_image(i, quiet=True) for i in p.init_images]
p.init_images = [i.convert('RGB') if i.mode != 'RGB' else i for i in p.init_images]
p.init_images = [i.convert('RGB') if i.mode != 'RGB' else i for i in p.init_images if i is not None]
if (sd_models.get_diffusers_task(model) == sd_models.DiffusersTaskType.TEXT_2_IMAGE or len(getattr(p, 'init_images', [])) == 0) and not is_img2img_model:
p.ops.append('txt2img')
if hasattr(p, 'width') and hasattr(p, 'height'):
@@ -262,6 +262,9 @@ def set_pipeline_args(p, model, prompts: list, negative_prompts: list, prompts_2
elif 'callback' in possible:
args['callback'] = diffusers_callback_legacy
if 'image' in kwargs and len(getattr(p, 'init_images', [])) == 0:
p.init_images = kwargs['image'] if isinstance(kwargs['image'], list) else [kwargs['image']]
# handle remaining args
for arg in kwargs:
if arg in possible: # add kwargs