feat: integrate MOSS-Image Edit with reference images and LoRA

README.md

@@ -164,6 +164,7 @@ models:
   - Boogu/Boogu-Image-0.1-Edit-Turbo
   - briaai/Fibo-2.0-dev
   - briaai/Fibo-2.0-edit-turbo
+  - chinoll/Moss-Image-Edit
   - darask0/MOSS-Image-color-depth-control
   - HiDream-ai/HiDream-I1-Dev
   - HiDream-ai/HiDream-I1-Fast

chain_injectors/moss_image_edit_injector.py

@@ -0,0 +1,228 @@
+import os
+
+try:
+    from utils.app_utils import ensure_file_downloaded
+except ImportError:
+    ensure_file_downloaded = None
+
+
+def create_node(assembler, class_type, title=None):
+    try:
+        node = assembler._get_node_template_from_api(class_type)
+    except Exception:
+        try:
+            node = assembler._get_node_template(class_type)
+        except Exception:
+            node = {
+                "inputs": {},
+                "class_type": class_type,
+                "_meta": {"title": title or class_type}
+            }
+    if title:
+        node.setdefault('_meta', {})['title'] = title
+    return node
+
+
+def inject(assembler, chain_definition, chain_items):
+    """
+    Injects MOSS-Image Edit (Reference Image Edit) into the workflow:
+    1. Loads the officially packaged Edit LoRA (moss_image_edit_lora_rank_64_bf16.safetensors)
+       onto the model path.
+    2. Loads and scales up to 10 reference images with ImageScaleToTotalPixels.
+    3. Replaces positive/negative prompt encoding with native TextEncodeMOSSImageEdit nodes,
+       which attach reference image latents and text tokens.
+    4. Connects patched model and conditioned positive/negative outputs to KSampler.
+    """
+    if not chain_items:
+        return
+
+    # Normalize chain_items: dict or list of dicts/strings
+    edit_format = "Moss-Image-Edit"
+    strength = 1.0
+    valid_images = []
+
+    if isinstance(chain_items, dict):
+        chain_items = [chain_items]
+
+    if isinstance(chain_items, list):
+        for item in chain_items:
+            if not item:
+                continue
+            if isinstance(item, dict):
+                raw_imgs = item.get('images')
+                if isinstance(raw_imgs, list):
+                    for img in raw_imgs:
+                        p = img.get('image') or img.get('filename') or img.get('path') if isinstance(img, dict) else img
+                        if p and isinstance(p, str) and p.strip():
+                            valid_images.append(p.strip())
+                elif isinstance(raw_imgs, str) and raw_imgs.strip():
+                    valid_images.append(raw_imgs.strip())
+
+                p = item.get('image') or item.get('filename') or item.get('path')
+                if p and isinstance(p, str) and p.strip():
+                    valid_images.append(p.strip())
+
+                if 'edit_format' in item:
+                    edit_format = item['edit_format']
+                if 'strength' in item:
+                    try:
+                        strength = float(item['strength'])
+                    except (ValueError, TypeError):
+                        pass
+            elif isinstance(item, str) and item.strip():
+                valid_images.append(item.strip())
+
+    if not valid_images:
+        return
+
+    valid_images = valid_images[:10]
+
+    lora_filename = "moss_image_edit_lora_rank_64_bf16.safetensors"
+    if ensure_file_downloaded:
+        try:
+            ensure_file_downloaded(lora_filename)
+        except Exception as e:
+            print(f"Warning: Failed to ensure '{lora_filename}' downloaded: {e}")
+
+    ksampler_name = chain_definition.get('ksampler_node', 'ksampler')
+    pos_prompt_name = chain_definition.get('pos_prompt_node', 'pos_prompt')
+    neg_prompt_name = chain_definition.get('neg_prompt_node', 'neg_prompt')
+    clip_loader_name = chain_definition.get('clip_loader_node', 'clip_loader')
+    vae_loader_name = chain_definition.get('vae_loader_node', 'vae_loader')
+
+    if ksampler_name not in assembler.node_map:
+        print(f"Warning: Target node '{ksampler_name}' for MOSS-Image Edit chain not found. Skipping.")
+        return
+
+    ksampler_id = assembler.node_map[ksampler_name]
+
+    if 'model' not in assembler.workflow[ksampler_id]['inputs']:
+        print(f"Warning: KSampler node '{ksampler_name}' is missing 'model' input. Skipping.")
+        return
+
+    current_model_connection = assembler.workflow[ksampler_id]['inputs']['model']
+
+    vae_connection = None
+    if vae_loader_name in assembler.node_map:
+        vae_connection = [assembler.node_map[vae_loader_name], 0]
+    else:
+        for node_id, node in assembler.workflow.items():
+            if isinstance(node, dict) and node.get('class_type') == 'VAELoader':
+                vae_connection = [node_id, 0]
+                break
+
+    if not vae_connection:
+        print("Warning: VAE connection not found for MOSS-Image Edit. Skipping.")
+        return
+
+    pos_prompt_id = assembler.node_map.get(pos_prompt_name)
+    neg_prompt_id = assembler.node_map.get(neg_prompt_name)
+
+    clip_connection = None
+    if clip_loader_name in assembler.node_map:
+        clip_connection = [assembler.node_map[clip_loader_name], 0]
+    elif pos_prompt_id and 'clip' in assembler.workflow.get(pos_prompt_id, {}).get('inputs', {}):
+        clip_connection = assembler.workflow[pos_prompt_id]['inputs']['clip']
+    else:
+        for node_id, node in assembler.workflow.items():
+            if isinstance(node, dict) and node.get('class_type') == 'CLIPLoader':
+                clip_connection = [node_id, 0]
+                break
+
+    if not clip_connection:
+        print("Warning: CLIP connection not found for MOSS-Image Edit. Skipping.")
+        return
+
+    # 1. Inject LoRA loader on model path if strength > 0
+    if strength > 0.0:
+        lora_loader_id = assembler._get_unique_id()
+        lora_loader_node = create_node(assembler, "LoraLoaderModelOnly", "Load LoRA (MOSS-Image Edit)")
+        lora_loader_node['inputs']['lora_name'] = lora_filename
+        lora_loader_node['inputs']['strength_model'] = strength
+        lora_loader_node['inputs']['model'] = current_model_connection
+        assembler.workflow[lora_loader_id] = lora_loader_node
+        current_model_connection = [lora_loader_id, 0]
+
+    megapixels = float(chain_definition.get('megapixels', 1.0))
+
+    # 2. Process Reference Images
+    scale_image_ids = []
+    for i, img_filename in enumerate(valid_images):
+        load_id = assembler._get_unique_id()
+        load_node = create_node(assembler, "LoadImage", f"Load Reference Image {i+1} (MOSS Edit)")
+        load_node['inputs']['image'] = img_filename
+        assembler.workflow[load_id] = load_node
+
+        scale_id = assembler._get_unique_id()
+        scale_node = create_node(assembler, "ImageScaleToTotalPixels", f"Scale Reference {i+1} (MOSS Edit)")
+        scale_node['inputs']['upscale_method'] = "lanczos"
+        scale_node['inputs']['megapixels'] = megapixels
+        scale_node['inputs']['resolution_steps'] = 1
+        scale_node['inputs']['image'] = [load_id, 0]
+        assembler.workflow[scale_id] = scale_node
+        scale_image_ids.append(scale_id)
+
+    # 3. Extract Prompt Texts
+    pos_text = ""
+    if pos_prompt_id and pos_prompt_id in assembler.workflow:
+        pos_text = assembler.workflow[pos_prompt_id]['inputs'].get('text', '') or assembler.workflow[pos_prompt_id]['inputs'].get('prompt', '')
+    elif hasattr(assembler, 'ui_values') and isinstance(assembler.ui_values, dict):
+        pos_text = assembler.ui_values.get('positive_prompt') or assembler.ui_values.get('prompt') or ''
+
+    neg_text = ""
+    if neg_prompt_id and neg_prompt_id in assembler.workflow:
+        neg_text = assembler.workflow[neg_prompt_id]['inputs'].get('text', '') or assembler.workflow[neg_prompt_id]['inputs'].get('prompt', '')
+    elif hasattr(assembler, 'ui_values') and isinstance(assembler.ui_values, dict):
+        neg_text = assembler.ui_values.get('negative_prompt') or assembler.ui_values.get('neg_prompt') or ''
+
+    # 4. Positive TextEncodeMOSSImageEdit
+    pos_edit_id = assembler._get_unique_id()
+    pos_edit_node = create_node(assembler, "TextEncodeMOSSImageEdit", "Text Encode MOSS-Image Edit")
+    pos_edit_node['inputs']['clip'] = clip_connection
+    pos_edit_node['inputs']['prompt'] = pos_text
+    pos_edit_node['inputs']['vae'] = vae_connection
+    pos_edit_node['inputs']['edit_format'] = edit_format
+    pos_edit_node['inputs']['image_strength'] = 1.0
+    for i, sc_id in enumerate(scale_image_ids):
+        pos_edit_node['inputs'][f"images.image_{i}"] = [sc_id, 0]
+    pos_edit_node['inputs'] = {k: v for k, v in pos_edit_node['inputs'].items() if v is not None}
+    assembler.workflow[pos_edit_id] = pos_edit_node
+
+    # 5. Negative TextEncodeMOSSImageEdit
+    neg_edit_id = assembler._get_unique_id()
+    neg_edit_node = create_node(assembler, "TextEncodeMOSSImageEdit", "Text Encode MOSS-Image Edit")
+    neg_edit_node['inputs']['clip'] = clip_connection
+    neg_edit_node['inputs']['prompt'] = neg_text
+    neg_edit_node['inputs']['vae'] = vae_connection
+    neg_edit_node['inputs']['edit_format'] = edit_format
+    neg_edit_node['inputs']['image_strength'] = 1.0
+    for i, sc_id in enumerate(scale_image_ids):
+        neg_edit_node['inputs'][f"images.image_{i}"] = [sc_id, 0]
+    neg_edit_node['inputs'] = {k: v for k, v in neg_edit_node['inputs'].items() if v is not None}
+    assembler.workflow[neg_edit_id] = neg_edit_node
+
+    # 6. Reconnect model and redirect conditioning references in workflow
+    assembler.workflow[ksampler_id]['inputs']['model'] = current_model_connection
+
+    # Redirect all existing connections that referenced pos_prompt_id / neg_prompt_id
+    if pos_prompt_id:
+        for node in assembler.workflow.values():
+            if isinstance(node, dict) and 'inputs' in node:
+                for k, v in node['inputs'].items():
+                    if isinstance(v, list) and len(v) == 2 and v[0] == pos_prompt_id:
+                        node['inputs'][k] = [pos_edit_id, 0]
+
+    if neg_prompt_id:
+        for node in assembler.workflow.values():
+            if isinstance(node, dict) and 'inputs' in node:
+                for k, v in node['inputs'].items():
+                    if isinstance(v, list) and len(v) == 2 and v[0] == neg_prompt_id:
+                        node['inputs'][k] = [neg_edit_id, 0]
+
+    # Clean up original CLIPTextEncode nodes if they are in workflow
+    if pos_prompt_id and pos_prompt_id in assembler.workflow:
+        del assembler.workflow[pos_prompt_id]
+    if neg_prompt_id and neg_prompt_id in assembler.workflow:
+        del assembler.workflow[neg_prompt_id]
+
+    print(f"MOSS-Image Edit injector applied with {len(valid_images)} reference image(s) (format: {edit_format}, strength: {strength}).")

core/pipelines/pipeline_input_processor.py

@@ -492,6 +492,20 @@ def process_pipeline_inputs(ui_inputs: Dict[str, Any], progress: gr.Progress, wo
                 'end_percent': end_percent
             })
 
+    moss_image_edit_data = ui_inputs.get('moss_image_edit_data', [])
+    active_moss_image_edit = []
+    if 'moss_image_edit' in enabled_chains and moss_image_edit_data:
+        for img in moss_image_edit_data:
+            if img:
+                if isinstance(img, str):
+                    active_moss_image_edit.append(img)
+                elif hasattr(img, 'save'):
+                    if not os.path.exists(INPUT_DIR): os.makedirs(INPUT_DIR)
+                    temp_path = os.path.join(INPUT_DIR, f"temp_moss_edit_{random.randint(1000, 9999)}.png")
+                    img.save(temp_path, "PNG")
+                    temp_files_to_clean.append(temp_path)
+                    active_moss_image_edit.append(os.path.basename(temp_path))
+
     return {
         "active_loras_for_gpu": active_loras_for_gpu,
         "active_loras_for_meta": active_loras_for_meta,
@@ -513,5 +527,6 @@ def process_pipeline_inputs(ui_inputs: Dict[str, Any], progress: gr.Progress, wo
         "active_ming_image_prompt_enhancer": active_ming_image_prompt_enhancer,
         "active_flux3_prompt_upsampler": active_flux3_prompt_upsampler,
         "active_moss_image_control": active_moss_image_control,
+        "active_moss_image_edit": active_moss_image_edit,
         "temp_files_to_clean": temp_files_to_clean
     }

core/pipelines/sd_image_pipeline.py

@@ -123,6 +123,7 @@ class SdImagePipeline(BasePipeline):
             active_ming_image_prompt_enhancer = processed.get("active_ming_image_prompt_enhancer", [])
             active_flux3_prompt_upsampler = processed.get("active_flux3_prompt_upsampler", [])
             active_moss_image_control = processed.get("active_moss_image_control", [])
+            active_moss_image_edit = processed.get("active_moss_image_edit", [])
 
             loras_string = f"LoRAs: [{', '.join(active_loras_for_meta)}]" if active_loras_for_meta else ""
             
@@ -184,6 +185,7 @@ class SdImagePipeline(BasePipeline):
                 "ming_image_prompt_enhancer_chain": active_ming_image_prompt_enhancer,
                 "flux3_prompt_upsampler_chain": active_flux3_prompt_upsampler,
                 "moss_image_control_chain": active_moss_image_control,
+                "moss_image_edit_chain": active_moss_image_edit,
                 "vae_chain": [ui_inputs.get('vae_name')] if (ui_inputs.get('vae_name') and 'vae' in enabled_chains) else [],
                 "hidream_o1_smoothing_chain": hidream_o1_smoothing_data,
                 "pid_chain": [ui_inputs.get('pid_settings', 'OFF')] if is_pid_enabled else [],

core/pipelines/workflow_recipes/_partials/conditioning/moss-image.yaml

@@ -76,6 +76,15 @@ dynamic_moss_image_control_chains:
     vae_loader_node: "vae_loader"
     megapixels: 1.0
 
+dynamic_moss_image_edit_chains:
+  moss_image_edit_chain:
+    ksampler_node: "ksampler"
+    clip_loader_node: "clip_loader"
+    vae_loader_node: "vae_loader"
+    pos_prompt_node: "pos_prompt"
+    neg_prompt_node: "neg_prompt"
+    megapixels: 1.0
+
 dynamic_pid_chains:
   pid_chain:
     ksampler_node: "ksampler"

core/settings.py

@@ -198,6 +198,7 @@ try:
     MAX_CONDITIONINGS = _constants.get('MAX_CONDITIONINGS', 10)
     MAX_CONTROLNETS = _constants.get('MAX_CONTROLNETS', 5)
     MAX_IPADAPTERS = _constants.get('MAX_IPADAPTERS', 5)
+    MAX_MOSS_IMAGE_EDITS = _constants.get('MAX_MOSS_IMAGE_EDITS', 10)
     LORA_SOURCE_CHOICES = _constants.get('LORA_SOURCE_CHOICES', ["Civitai", "Hugging Face", "Upload File"])
     RESOLUTION_MAP = _constants.get('RESOLUTION_MAP', {})
     MULTIPLIERS_MAP = _constants.get('MULTIPLIERS_MAP', {})
@@ -207,7 +208,7 @@ try:
     MODEL_DEFAULTS_CONFIG = load_model_defaults()
 except Exception as e:
     print(f"FATAL: Could not load constants from YAML. Error: {e}")
-    MAX_LORAS, MAX_EMBEDDINGS, MAX_CONDITIONINGS, MAX_CONTROLNETS, MAX_IPADAPTERS = 10, 5, 10, 5, 5
+    MAX_LORAS, MAX_EMBEDDINGS, MAX_CONDITIONINGS, MAX_CONTROLNETS, MAX_IPADAPTERS, MAX_MOSS_IMAGE_EDITS = 10, 5, 10, 5, 5, 10
     LORA_SOURCE_CHOICES = ["Civitai", "Hugging Face", "Upload File"]
     RESOLUTION_MAP = {}
     MULTIPLIERS_MAP = {}

mcp_tools/common.py

@@ -447,6 +447,7 @@ def _execute_imagegen_pipeline(task_id: str, params: dict):
             qwen_image_edit_data = []
             hidream_o1_reference_data = []
             sensenova_reference_data = []
+            moss_image_edit_data = []
             cond_prompts = []
             cond_widths = []
             cond_heights = []
@@ -700,6 +701,16 @@ def _execute_imagegen_pipeline(task_id: str, params: dict):
                     img = _parse_image_param(item.get("image"))
                     if img:
                         sensenova_reference_data.append(img)
+                elif itype == "moss_image_edit":
+                    if "images" in item and isinstance(item["images"], list):
+                        for img_item in item["images"]:
+                            p_img = _parse_image_param(img_item)
+                            if p_img:
+                                moss_image_edit_data.append(p_img)
+                    elif "image" in item:
+                        p_img = _parse_image_param(item["image"])
+                        if p_img:
+                            moss_image_edit_data.append(p_img)
                 elif itype == "vae":
                     v_source = item.get("source", item.get("vae_source", "Civitai"))
                     v_val = item.get("vae_value", item.get("value", item.get("vae_id", item.get("vae_name", ""))))
@@ -710,6 +721,7 @@ def _execute_imagegen_pipeline(task_id: str, params: dict):
             if lora_data: ui_inputs["lora_data"] = lora_data
             if embedding_data: ui_inputs["embedding_data"] = embedding_data
             if controlnet_data: ui_inputs["controlnet_data"] = controlnet_data
+            if moss_image_edit_data: ui_inputs["moss_image_edit_data"] = moss_image_edit_data
 
             if ipadapter_images:
                 preset = ipadapter_global_preset or "STANDARD (medium strength)"
@@ -839,6 +851,25 @@ def _execute_imagegen_pipeline(task_id: str, params: dict):
                 except (ValueError, TypeError):
                     pass
 
+        moss_edit_val = params.get("moss_image_edit")
+        if moss_edit_val:
+            m_imgs = []
+            if isinstance(moss_edit_val, dict):
+                raw_imgs = moss_edit_val.get("images", [])
+                if isinstance(raw_imgs, list):
+                    for img_item in raw_imgs:
+                        p_img = _parse_image_param(img_item)
+                        if p_img: m_imgs.append(p_img)
+                elif "image" in moss_edit_val:
+                    p_img = _parse_image_param(moss_edit_val["image"])
+                    if p_img: m_imgs.append(p_img)
+            elif isinstance(moss_edit_val, list):
+                for img_item in moss_edit_val:
+                    p_img = _parse_image_param(img_item)
+                    if p_img: m_imgs.append(p_img)
+            if m_imgs:
+                ui_inputs["moss_image_edit_data"] = m_imgs
+
         _TASKS_DB[task_id]["progress"] = 50
 
         # Execute Pipeline

ui/events/chain_handlers.py

@@ -689,4 +689,39 @@ def create_krea2_reference_edit_event_handlers(prefix, ui_components):
     add_outputs = [count_state, add_button, del_button] + ref_rows
     del_outputs = [count_state, add_button, del_button] + ref_rows + images
     add_button.click(fn=add_ref_row, inputs=[count_state], outputs=add_outputs, show_progress=False)
-    del_button.click(fn=del_ref_row, inputs=[count_state], outputs=del_outputs, show_progress=False)
+    del_button.click(fn=del_ref_row, inputs=[count_state], outputs=del_outputs, show_progress=False)
+
+
+def create_moss_image_edit_event_handlers(prefix, ui_components):
+    ref_rows = ui_components.get(f'moss_image_edit_rows_{prefix}')
+    if not ref_rows: return
+    count_state = ui_components[f'moss_image_edit_count_state_{prefix}']
+    add_button = ui_components[f'add_moss_image_edit_button_{prefix}']
+    del_button = ui_components[f'delete_moss_image_edit_button_{prefix}']
+    images = ui_components[f'moss_image_edit_images_{prefix}']
+    max_count = len(ref_rows)
+
+    def add_ref_row(c):
+        c += 1
+        return {
+            count_state: c,
+            ref_rows[c - 1]: gr.update(visible=True),
+            add_button: gr.update(visible=c < max_count),
+            del_button: gr.update(visible=True),
+        }
+
+    def del_ref_row(c):
+        c -= 1
+        return {
+            count_state: c,
+            ref_rows[c]: gr.update(visible=False),
+            images[c]: None,
+            add_button: gr.update(visible=True),
+            del_button: gr.update(visible=c > 1),
+        }
+
+    add_outputs = [count_state, add_button, del_button] + ref_rows
+    del_outputs = [count_state, add_button, del_button] + ref_rows + images
+    add_button.click(fn=add_ref_row, inputs=[count_state], outputs=add_outputs, show_progress=False)
+    del_button.click(fn=del_ref_row, inputs=[count_state], outputs=del_outputs, show_progress=False)
+

ui/events/change_handlers.py

@@ -15,7 +15,7 @@ from .config_loaders import (
     load_ipadapter_config
 )
 
-def make_update_fn(m_comp, cat_comp, ar_comp, width_comp, height_comp, cn_types, cn_series, cn_filepaths, ipa_preset, lora_acc, cn_acc, ipa_acc, sd3_ipa_acc, flux1_ipa_acc, style_acc, embed_acc, cond_acc, ref_latent_acc, hidream_o1_ref_acc, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp, pid_acc=None, vae_acc=None, joyai_ref_acc=None, krea2_identity_edit_acc=None, krea2_reference_edit_acc=None, qwen_image_edit_acc=None, ref_img_acc=None, sensenova_ref_acc=None, task_type=None, type_comp=None, qwen_pe_acc=None, ming_pe_acc=None, flux3_pe_acc=None, moss_image_control_acc=None):
+def make_update_fn(m_comp, cat_comp, ar_comp, width_comp, height_comp, cn_types, cn_series, cn_filepaths, ipa_preset, lora_acc, cn_acc, ipa_acc, sd3_ipa_acc, flux1_ipa_acc, style_acc, embed_acc, cond_acc, ref_latent_acc, hidream_o1_ref_acc, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp, pid_acc=None, vae_acc=None, joyai_ref_acc=None, krea2_identity_edit_acc=None, krea2_reference_edit_acc=None, qwen_image_edit_acc=None, ref_img_acc=None, sensenova_ref_acc=None, task_type=None, type_comp=None, qwen_pe_acc=None, ming_pe_acc=None, flux3_pe_acc=None, moss_image_control_acc=None, moss_image_edit_acc=None):
     def update_fn(*args):
         arch = args[0]
         category = args[1]
@@ -88,6 +88,7 @@ def make_update_fn(m_comp, cat_comp, ar_comp, width_comp, height_comp, cn_types,
         if ming_pe_acc: updates[ming_pe_acc] = gr.update(visible=('ming_image_prompt_enhancer' in enabled_chains))
         if flux3_pe_acc: updates[flux3_pe_acc] = gr.update(visible=('flux3_prompt_upsampler' in enabled_chains))
         if moss_image_control_acc: updates[moss_image_control_acc] = gr.update(visible=('moss_image_control' in enabled_chains))
+        if moss_image_edit_acc: updates[moss_image_edit_acc] = gr.update(visible=('moss_image_edit' in enabled_chains))
         
         if ar_comp:
             res_key = arch_model_type
@@ -132,7 +133,7 @@ def make_update_fn(m_comp, cat_comp, ar_comp, width_comp, height_comp, cn_types,
     return update_fn
 
 
-def make_model_change_fn(cat_comp_ref, ar_comp, width_comp, height_comp, cn_types, cn_series, cn_filepaths, arch_comp_ref, ipa_preset, lora_acc, cn_acc, ipa_acc, sd3_ipa_acc, flux1_ipa_acc, style_acc, embed_acc, cond_acc, ref_latent_acc, hidream_o1_ref_acc, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp, pid_acc=None, vae_acc=None, joyai_ref_acc=None, krea2_identity_edit_acc=None, krea2_reference_edit_acc=None, qwen_image_edit_acc=None, ref_img_acc=None, sensenova_ref_acc=None, task_type=None, type_comp=None, qwen_pe_acc=None, ming_pe_acc=None, flux3_pe_acc=None, moss_image_control_acc=None):
+def make_model_change_fn(cat_comp_ref, ar_comp, width_comp, height_comp, cn_types, cn_series, cn_filepaths, arch_comp_ref, ipa_preset, lora_acc, cn_acc, ipa_acc, sd3_ipa_acc, flux1_ipa_acc, style_acc, embed_acc, cond_acc, ref_latent_acc, hidream_o1_ref_acc, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp, pid_acc=None, vae_acc=None, joyai_ref_acc=None, krea2_identity_edit_acc=None, krea2_reference_edit_acc=None, qwen_image_edit_acc=None, ref_img_acc=None, sensenova_ref_acc=None, task_type=None, type_comp=None, qwen_pe_acc=None, ming_pe_acc=None, flux3_pe_acc=None, moss_image_control_acc=None, moss_image_edit_acc=None):
     def change_fn(*args):
         model_name = args[0]
         idx = 1
@@ -213,6 +214,7 @@ def make_model_change_fn(cat_comp_ref, ar_comp, width_comp, height_comp, cn_type
         if ming_pe_acc: updates[ming_pe_acc] = gr.update(visible=('ming_image_prompt_enhancer' in enabled_chains))
         if flux3_pe_acc: updates[flux3_pe_acc] = gr.update(visible=('flux3_prompt_upsampler' in enabled_chains))
         if moss_image_control_acc: updates[moss_image_control_acc] = gr.update(visible=('moss_image_control' in enabled_chains))
+        if moss_image_edit_acc: updates[moss_image_edit_acc] = gr.update(visible=('moss_image_edit' in enabled_chains))
         
         if ar_comp:
             res_key = arch_model_type

ui/events/main.py

@@ -15,7 +15,8 @@ from .chain_handlers import (
     create_krea2_identity_edit_event_handlers,
     create_krea2_reference_edit_event_handlers,
     create_qwen_image_edit_event_handlers,
-    create_reference_image_event_handlers
+    create_reference_image_event_handlers,
+    create_moss_image_edit_event_handlers
 )
 from .change_handlers import (
     make_update_fn,
@@ -66,6 +67,7 @@ def attach_event_handlers(ui_components, demo):
         ming_pe_accordion = ui_components.get(f'ming_image_prompt_enhancer_accordion_{prefix}')
         flux3_pe_accordion = ui_components.get(f'flux3_prompt_upsampler_accordion_{prefix}')
         moss_control_accordion = ui_components.get(f'moss_image_control_accordion_{prefix}')
+        moss_edit_accordion = ui_components.get(f'moss_image_edit_accordion_{prefix}')
         
         ipa_preset_list = ui_components.get(f'ipadapter_final_preset_{prefix}')
         
@@ -105,6 +107,7 @@ def attach_event_handlers(ui_components, demo):
             if ming_pe_accordion: outputs.append(ming_pe_accordion)
             if flux3_pe_accordion: outputs.append(flux3_pe_accordion)
             if moss_control_accordion: outputs.append(moss_control_accordion)
+            if moss_edit_accordion: outputs.append(moss_edit_accordion)
             if ipa_preset_list: outputs.append(ipa_preset_list)
             
             outputs.extend(valid_extra_comps)
@@ -116,7 +119,8 @@ def attach_event_handlers(ui_components, demo):
                 ref_latent_accordion, hidream_o1_ref_accordion, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp,
                 pid_acc=pid_accordion, vae_acc=vae_accordion, joyai_ref_acc=joyai_ref_accordion, krea2_identity_edit_acc=krea2_identity_edit_accordion, krea2_reference_edit_acc=krea2_reference_edit_accordion, qwen_image_edit_acc=qwen_image_edit_accordion, ref_img_acc=ref_img_accordion, sensenova_ref_acc=sensenova_ref_accordion,
                 task_type=task_type, type_comp=type_comp, qwen_pe_acc=qwen_pe_accordion, ming_pe_acc=ming_pe_accordion, flux3_pe_acc=flux3_pe_accordion,
-                moss_image_control_acc=moss_control_accordion
+                moss_image_control_acc=moss_control_accordion,
+                moss_image_edit_acc=moss_edit_accordion
             )
             inputs = [arch_comp, cat_comp]
             if aspect_ratio_comp:
@@ -154,6 +158,7 @@ def attach_event_handlers(ui_components, demo):
             if ming_pe_accordion: outputs2.append(ming_pe_accordion)
             if flux3_pe_accordion: outputs2.append(flux3_pe_accordion)
             if moss_control_accordion: outputs2.append(moss_control_accordion)
+            if moss_edit_accordion: outputs2.append(moss_edit_accordion)
             if ipa_preset_list: outputs2.append(ipa_preset_list)
             
             outputs2.extend(valid_extra_comps)
@@ -170,7 +175,8 @@ def attach_event_handlers(ui_components, demo):
                     ref_latent_accordion, hidream_o1_ref_accordion, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp,
                     pid_acc=pid_accordion, vae_acc=vae_accordion, joyai_ref_acc=joyai_ref_accordion, krea2_identity_edit_acc=krea2_identity_edit_accordion, krea2_reference_edit_acc=krea2_reference_edit_accordion, qwen_image_edit_acc=qwen_image_edit_accordion, ref_img_acc=ref_img_accordion, sensenova_ref_acc=sensenova_ref_accordion,
                     task_type=task_type, type_comp=type_comp, qwen_pe_acc=qwen_pe_accordion, ming_pe_acc=ming_pe_accordion, flux3_pe_acc=flux3_pe_accordion,
-                    moss_image_control_acc=moss_control_accordion
+                    moss_image_control_acc=moss_control_accordion,
+                    moss_image_edit_acc=moss_edit_accordion
                 )
                 if type_comp:
                     inputs2.append(type_comp)
@@ -192,6 +198,7 @@ def attach_event_handlers(ui_components, demo):
         create_krea2_reference_edit_event_handlers(prefix, ui_components)
         create_qwen_image_edit_event_handlers(prefix, ui_components)
         create_reference_image_event_handlers(prefix, ui_components)
+        create_moss_image_edit_event_handlers(prefix, ui_components)
         create_run_event(prefix, task_type, ui_components)
 
     # Dynamic Type toggle event for unified imagegen UI
@@ -262,6 +269,7 @@ def attach_event_handlers(ui_components, demo):
                 f'ming_image_prompt_enhancer_accordion_{prefix}': 'ming_image_prompt_enhancer',
                 f'flux3_prompt_upsampler_accordion_{prefix}': 'flux3_prompt_upsampler',
                 f'moss_image_control_accordion_{prefix}': 'moss_image_control',
+                f'moss_image_edit_accordion_{prefix}': 'moss_image_edit',
                 f'pid_accordion_{prefix}': 'pid',
                 f'vae_accordion_{prefix}': 'vae',
             }
@@ -303,6 +311,7 @@ def attach_event_handlers(ui_components, demo):
             'ming_image_prompt_enhancer_accordion_imagegen',
             'flux3_prompt_upsampler_accordion_imagegen',
             'moss_image_control_accordion_imagegen',
+            'moss_image_edit_accordion_imagegen',
             'pid_accordion_imagegen', 'vae_accordion_imagegen',
         ]
         for key in accordion_keys:

ui/events/run_handlers.py

@@ -119,6 +119,7 @@ def create_run_event(prefix: str, task_type: str = None, ui_components: dict = N
     krea2_reference_edit_data_components = ui_components.get(f'all_krea2_reference_edit_components_flat_{prefix}', [])
     qwen_image_edit_data_components = ui_components.get(f'all_qwen_image_edit_components_flat_{prefix}', [])
     reference_image_data_components = ui_components.get(f'all_reference_image_components_flat_{prefix}', [])
+    moss_image_edit_data_components = ui_components.get(f'all_moss_image_edit_components_flat_{prefix}', [])
     
     run_inputs_map['vae_source'] = ui_components.get(f'vae_source_{prefix}')
     run_inputs_map['vae_id'] = ui_components.get(f'vae_id_{prefix}')
@@ -129,7 +130,8 @@ def create_run_event(prefix: str, task_type: str = None, ui_components: dict = N
     all_chains = [
         lora_data_components, controlnet_data_components, ipadapter_data_components,
         sd3_ipadapter_data_components, flux1_ipadapter_data_components, style_data_components,
-        embedding_data_components, conditioning_data_components, reference_latent_data_components, hidream_o1_reference_data_components, sensenova_reference_data_components, joyai_reference_data_components, krea2_identity_edit_data_components, krea2_reference_edit_data_components, qwen_image_edit_data_components, reference_image_data_components
+        embedding_data_components, conditioning_data_components, reference_latent_data_components, hidream_o1_reference_data_components, sensenova_reference_data_components, joyai_reference_data_components, krea2_identity_edit_data_components, krea2_reference_edit_data_components, qwen_image_edit_data_components, reference_image_data_components,
+        moss_image_edit_data_components
     ]
     for chain in all_chains:
         if chain:
@@ -162,6 +164,7 @@ def create_run_event(prefix: str, task_type: str = None, ui_components: dict = N
         assign_chain_data('krea2_reference_edit_data', krea2_reference_edit_data_components)
         assign_chain_data('qwen_image_edit_data', qwen_image_edit_data_components)
         assign_chain_data('reference_image_data', reference_image_data_components)
+        assign_chain_data('moss_image_edit_data', moss_image_edit_data_components)
 
         if 'ui_type' in ui_dict:
             ui_dict['task_type'] = TYPE_TASK_MAP.get(ui_dict['ui_type'], 'txt2img')

ui/imagegen_ui.py

@@ -10,7 +10,8 @@ from .shared.ui_components import (
     create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
     create_reference_latent_ui, create_hidream_o1_reference_ui, create_sensenova_reference_ui, create_joyai_reference_ui, create_reference_image_ui, create_krea2_identity_edit_ui, create_krea2_reference_edit_ui, create_qwen_image_edit_ui,
     create_pid_ui, create_qwen_image_2_1_prompt_enhancer_ui, create_ming_image_prompt_enhancer_ui,
-    create_flux3_prompt_upsampler_ui, create_moss_image_control_ui
+    create_flux3_prompt_upsampler_ui, create_moss_image_control_ui,
+    create_moss_image_edit_ui
 )
 
 default_vals = MODEL_DEFAULTS_CONFIG.get('Default', {})
@@ -193,6 +194,7 @@ def create_ui():
             components.update(create_flux3_prompt_upsampler_ui(prefix))
             components.update(create_vae_override_ui(prefix))
             components.update(create_moss_image_control_ui(prefix))
+            components.update(create_moss_image_edit_ui(prefix))
             components.update(create_pid_ui(prefix))
         components[f'accordion_wrapper_{prefix}'] = accordion_wrapper
         

ui/shared/ui_components.py

@@ -2,7 +2,7 @@ import gradio as gr
 from comfy_integration.nodes import SAMPLER_CHOICES, SCHEDULER_CHOICES
 from core.settings import (
     MAX_LORAS, LORA_SOURCE_CHOICES, MAX_EMBEDDINGS, MAX_CONDITIONINGS,
-    MAX_CONTROLNETS, MAX_IPADAPTERS, RESOLUTION_MAP, ARCHITECTURES_CONFIG,
+    MAX_CONTROLNETS, MAX_IPADAPTERS, MAX_MOSS_IMAGE_EDITS, RESOLUTION_MAP, ARCHITECTURES_CONFIG,
     MODEL_MAP_CHECKPOINT, MODEL_TYPE_MAP, FEATURES_CONFIG, ARCH_CATEGORIES_MAP,
     VAE_DIR, MODEL_DEFAULTS_CONFIG
 )
@@ -912,4 +912,36 @@ def create_flux3_prompt_upsampler_ui(prefix: str):
             )
 
     return components
+
+
+def create_moss_image_edit_ui(prefix: str, max_units=None):
+    if max_units is None:
+        max_units = MAX_MOSS_IMAGE_EDITS
+    components = {}
+    key = lambda name: f"{name}_{prefix}"
+
+    with gr.Accordion("MOSS-Image Edit Settings", open=False, visible=('moss_image_edit' in default_enabled_chains)) as ref_accordion:
+        components[key('moss_image_edit_accordion')] = ref_accordion
+        gr.Markdown("💡 **Tip:** For multimodal models, this feature enables powerful editing and combining capabilities. In txt2img mode, adding a single reference image performs an Image Edit, while adding multiple images performs an Image Combine. Uses [chinoll/Moss-Image-Edit](https://huggingface.co/chinoll/Moss-Image-Edit).")
+        
+        ref_image_groups = []
+        ref_image_inputs = []
+        with gr.Row():
+            for i in range(max_units):
+                with gr.Column(visible=(i < 1), min_width=160) as img_col:
+                    img_comp = gr.Image(type="pil", label=f"Ref. {i+1}", sources=["upload"], height=150)
+                    ref_image_groups.append(img_col)
+                    ref_image_inputs.append(img_comp)
+        
+        components[key('moss_image_edit_rows')] = ref_image_groups
+        components[key('moss_image_edit_images')] = ref_image_inputs
+
+        with gr.Row():
+            components[key('add_moss_image_edit_button')] = gr.Button("✚ Add Reference Image")
+            components[key('delete_moss_image_edit_button')] = gr.Button("➖ Delete Reference Image", visible=False)
+        components[key('moss_image_edit_count_state')] = gr.State(1)
+        
+        components[key('all_moss_image_edit_components_flat')] = ref_image_inputs
+
+    return components
 

yaml/chain_features.yaml

@@ -557,3 +557,18 @@ moss_image_control:
         minimum: 0.0
         maximum: 1.0
         description: "Sampling percentage at which control ends (default 0.5, use 0.5 for 3D mannequin)."
+
+moss_image_edit:
+  chains: moss_image_edit
+  display_name: "MOSS-Image Edit"
+  description: "Edits images or generates variants using MOSS-Image Edit LoRA (officially packaged by Comfy-Org as moss_image_edit_lora_rank_64_bf16.safetensors) and native TextEncodeMOSSImageEdit node."
+  max_count: 10
+  usage_guideline: "Upload 1-10 reference images and specify prompt. Automatically uses the official MOSS-Image Edit LoRA layout."
+  parameters_schema:
+    type: object
+    properties:
+      images:
+        type: array
+        items:
+          type: string
+        description: "List of 1 to 10 reference images encoded as Base64 Data URI, HTTP/HTTPS URL, or local file path."

yaml/constants.yaml

@@ -4,6 +4,7 @@ MAX_IPADAPTERS: 5
 MAX_EMBEDDINGS: 5
 MAX_CONDITIONINGS: 10
 MAX_REFERENCE_LATENTS: 10
+MAX_MOSS_IMAGE_EDITS: 10
 LORA_SOURCE_CHOICES: ["Civitai", "Hugging Face", "Upload File"]
 
 RESOLUTION_MAP:

yaml/file_list.yaml

@@ -892,6 +892,11 @@ file:
       source: "hf"
       repo_id: "darask0/MOSS-Image-color-depth-control"
       repository_file_path: "moss-control.safetensors"
+    # MOSS-Image Edit
+    - filename: "moss_image_edit_lora_rank_64_bf16.safetensors"
+      source: "hf"
+      repo_id: "Comfy-Org/MOSS-Image"
+      repository_file_path: "loras/moss_image_edit_lora_rank_64_bf16.safetensors"
     # Krea2 Identity Edit
     - filename: "krea2_identity_edit_v1_2.safetensors"
       source: "hf"

yaml/model_architecture_features.yaml

@@ -87,6 +87,7 @@ moss-image:
   enabled_chains:
   - lora
   - moss_image_control
+  - moss_image_edit
   - pid
 
 ming-image:

yaml/task_features.yaml

@@ -26,6 +26,7 @@ txt2img:
   - ming_image_prompt_enhancer
   - flux3_prompt_upsampler
   - moss_image_control
+  - moss_image_edit
   - pid
   - vae
 
@@ -52,6 +53,7 @@ img2img:
   - ming_image_prompt_enhancer
   - flux3_prompt_upsampler
   - moss_image_control
+  - moss_image_edit
   - vae
 
 inpaint:
@@ -77,6 +79,7 @@ inpaint:
   - ming_image_prompt_enhancer
   - flux3_prompt_upsampler
   - moss_image_control
+  - moss_image_edit
   - vae
 
 outpaint:
@@ -102,6 +105,7 @@ outpaint:
   - ming_image_prompt_enhancer
   - flux3_prompt_upsampler
   - moss_image_control
+  - moss_image_edit
   - vae
 
 hires_fix:
@@ -127,4 +131,5 @@ hires_fix:
   - ming_image_prompt_enhancer
   - flux3_prompt_upsampler
   - moss_image_control
+  - moss_image_edit
   - vae