add llada editing

Signed-off-by: Vladimir Mandic <mandic00@live.com>
2026-09-19 09:14:35 +02:00 · 2026-09-05 13:36:18 +02:00
parent cb0aea8eb5
commit 9b06cfef80
3 changed files with 13 additions and 3 deletions
@@ -110,6 +110,11 @@ def task_specific_kwargs(p, model):
                'height': p.height,
                'input_images': [p.init_images], # omnigen expects list-of-lists
            }
+        elif model_cls == 'LLaDAImagePipeline':
+            task_args = {
+                'generation_mode': 'editing',
+                'image': p.init_images[0],
+            }
    elif task_type == sd_models.DiffusersTaskType.INSTRUCT and len(getattr(p, 'init_images', [])) > 0:
        p.ops.append('instruct')
        task_args = {