redo flux2

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-04-26 08:38:55 -04:00
parent 131c9043f6
commit 15ea60b569
5 changed files with 29 additions and 14 deletions
+2 -2
View File
@@ -284,8 +284,8 @@ def set_pipeline_args(p, model, prompts:list, negative_prompts:list, prompts_2:t
args["prior_guidance_scale"] = p.cfg_scale
if 'decoder_guidance_scale' in possible:
args["decoder_guidance_scale"] = p.image_cfg_scale
if 'Flex' in model.__class__.__name__:
if p.init_images is not None and len(p.init_images) > 0:
if 'Flex2' in model.__class__.__name__:
if len(getattr(p, 'init_images', [])) > 0:
args['inpaint_image'] = p.init_images[0] if isinstance(p.init_images, list) else p.init_images
args['inpaint_mask'] = Image.new('L', args['inpaint_image'].size, 1)
args['control_image'] = args['inpaint_image'].convert('L').convert('RGB') # will be interpreted as depth