feat: add MOSS-Image, FLUX.3 Base, fix PiD cleanup and update dependencies

README.md

@@ -56,6 +56,7 @@ models:
   - Comfy-Org/Lumina_Image_2.0_Repackaged
   - Comfy-Org/Mage-Flow
   - Comfy-Org/Ming-Image
+  - Comfy-Org/MOSS-Image
   - Comfy-Org/NewBie-image-Exp0.1_repackaged
   - Comfy-Org/Omnigen2_ComfyUI_repackaged
   - Comfy-Org/Ovis-Image
@@ -154,6 +155,7 @@ models:
   - black-forest-labs/FLUX.1-Krea-dev
   - black-forest-labs/FLUX.1-schnell
   - black-forest-labs/flux-3-image-klein
+  - black-forest-labs/flux-3-image-klein-base
   - black-forest-labs/flux-3-image-klein-pup
   - Boogu/Boogu-Image-0.1-Turbo
   - Boogu/Boogu-Image-0.1-Base
@@ -184,6 +186,8 @@ models:
   - nvidia/PiD
   - nvidia/PixelDiT-1300M-1024px
   - OmniGen2/OmniGen2
+  - OpenMOSS-Team/MOSS-Image-Base
+  - OpenMOSS-Team/MOSS-Image-SFT
   - Qwen/Qwen3.8-27B
   - Qwen/Qwen-Image
   - Qwen/Qwen-Image-2512

chain_injectors/pid_injector.py

@@ -290,6 +290,9 @@ def inject(assembler, chain_definition, chain_items):
                         if input_val[0] == original_vae_decode_id:
                             node_data['inputs'][input_name] = [pid_vae_decode_id, 0]
 
+    if original_vae_decode_id in assembler.workflow:
+        del assembler.workflow[original_vae_decode_id]
+
     is_vae_loader_referenced = False
     if original_vae_loader_id:
         for node_id, node_data in assembler.workflow.items():
@@ -306,7 +309,4 @@ def inject(assembler, chain_definition, chain_items):
     if original_vae_loader_id in assembler.workflow and not is_vae_loader_referenced:
         del assembler.workflow[original_vae_loader_id]
 
-    if original_vae_decode_id in assembler.workflow:
-        del assembler.workflow[original_vae_decode_id]
-
     print("[PiD Injector] Successfully injected PiD pipeline and replaced VAE decode/loader.")

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

@@ -0,0 +1,82 @@
+nodes:
+  pos_prompt:
+    class_type: CLIPTextEncode
+    title: "moss-image Positive"
+  neg_prompt:
+    class_type: CLIPTextEncode
+    title: "moss-image Negative"
+  unet_loader:
+    class_type: UNETLoader
+    title: "Load Diffusion Model"
+    params:
+      weight_dtype: "default"
+  slg:
+    class_type: SkipLayerGuidanceDiTSimple
+    title: "SkipLayerGuidanceDiTSimple"
+    params:
+      double_layers: ""
+      single_layers: "22"
+      start_percent: 0.0
+      end_percent: 0.5
+  clip_loader:
+    class_type: CLIPLoader
+    title: "Load CLIP"
+    params:
+      type: "moss_image"
+      device: "default"
+  vae_loader:
+    class_type: VAELoader
+    title: "Load VAE"
+
+connections:
+  - from: "unet_loader:0"
+    to: "slg:model"
+  - from: "slg:0"
+    to: "ksampler:model"
+  - from: "clip_loader:0"
+    to: "pos_prompt:clip"
+  - from: "clip_loader:0"
+    to: "neg_prompt:clip"
+  - from: "pos_prompt:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+  - from: "vae_loader:0"
+    to: "vae_decode:vae"
+  - from: "vae_loader:0"
+    to: "vae_encode:vae"
+
+dynamic_lora_chains:
+  lora_chain:
+    template: "LoraLoader"
+    output_map:
+      "unet_loader:0": "model"
+      "clip_loader:0": "clip"
+    input_map:
+      "model": "model"
+      "clip": "clip"
+    end_input_map:
+      "model": ["slg:model"]
+      "clip": ["pos_prompt:clip", "neg_prompt:clip"]
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    ksampler_node: "ksampler"
+    clip_source: "clip_loader:0"
+
+dynamic_vae_chains:
+  vae_chain:
+    targets:
+      - "vae_decode:vae"
+      - "vae_encode:vae"
+
+dynamic_pid_chains:
+  pid_chain:
+    ksampler_node: "ksampler"
+
+ui_map:
+  positive_prompt: "pos_prompt:text"
+  negative_prompt: "neg_prompt:text"
+  unet_name: "unet_loader:unet_name"
+  clip_name: "clip_loader:clip_name"
+  vae_name: "vae_loader:vae_name"

requirements.txt

@@ -1,5 +1,5 @@
 comfyui-frontend-package==1.55.17
-comfyui-workflow-templates==0.11.85
+comfyui-workflow-templates==0.11.86
 comfyui-embedded-docs==0.5.14
 torch
 torchsde
@@ -21,7 +21,7 @@ alembic
 SQLAlchemy>=2.0.0
 filelock
 av>=17.0.0
-comfy-kitchen==0.2.37
+comfy-kitchen==0.2.38
 comfy-aimdo==0.5.6
 requests
 simpleeval>=1.0.0

yaml/constants.yaml

@@ -111,6 +111,14 @@ RESOLUTION_MAP:
     "3:4 (Classic Portrait)": [896, 1152]
     "3:2 (Photography)": [1216, 832]
     "2:3 (Photography Portrait)": [832, 1216]
+  moss-image:
+    "1:1 (Square)": [1024, 1024]
+    "16:9 (Landscape)": [1344, 768]
+    "9:16 (Portrait)": [768, 1344]
+    "4:3 (Classic)": [1152, 896]
+    "3:4 (Classic Portrait)": [896, 1152]
+    "3:2 (Photography)": [1216, 832]
+    "2:3 (Photography Portrait)": [832, 1216]
   ming-image:
     "1:1 (Square)": [1024, 1024]
     "16:9 (Landscape)": [1344, 768]

yaml/file_list.yaml

@@ -406,14 +406,27 @@ file:
       repository_file_path: "control_v11u_sd15_tile_fp16.safetensors"
   diffusion_models: 
     # Flux-3
-    - filename: "flux3_image_klein_nvfp4.safetensors"
+    - filename: "flux3-image-klein-nvfp4.safetensors"
       source: "hf"
       repo_id: "black-forest-labs/flux-3-image-klein-comfy"
-      repository_file_path: "diffusion_models/flux3_image_klein_nvfp4.safetensors"
+      repository_file_path: "diffusion_models/flux3-image-klein-nvfp4.safetensors"
     - filename: "flux3_image_klein_sd_nvfp4.safetensors"
       source: "hf"
       repo_id: "black-forest-labs/flux-3-image-klein-comfy"
       repository_file_path: "diffusion_models/flux3_image_klein_sd_nvfp4.safetensors"
+    - filename: "flux3-image-klein-base-nvfp4.safetensors"
+      source: "hf"
+      repo_id: "black-forest-labs/flux-3-image-klein-comfy"
+      repository_file_path: "diffusion_models/flux3-image-klein-base-nvfp4.safetensors"
+    # MOSS-Image
+    - filename: "moss_image_base_int8_convrot.safetensors"
+      source: "hf"
+      repo_id: "Comfy-Org/MOSS-Image"
+      repository_file_path: "diffusion_models/moss_image_base_int8_convrot.safetensors"
+    - filename: "moss_image_sft_int8_convrot.safetensors"
+      source: "hf"
+      repo_id: "Comfy-Org/MOSS-Image"
+      repository_file_path: "diffusion_models/moss_image_sft_int8_convrot.safetensors"
     # Krea-2
     - filename: "krea2_turbo_nvfp4.safetensors"
       source: "hf"
@@ -938,6 +951,11 @@ file:
       repo_id: "Comfy-Org/Qwen-Image-2.1"
       repository_file_path: "model_patches/qwen_image_2.1_fun_controlnet_union_int8_convrot.safetensors"
   text_encoders: 
+    # MOSS-Image
+    - filename: "qwen3.5_9b_int8_convrot.safetensors"
+      source: "hf"
+      repo_id: "Comfy-Org/MOSS-Image"
+      repository_file_path: "text_encoders/qwen3.5_9b_int8_convrot.safetensors"
     # Krea-2
     - filename: "qwen3vl_4b_fp8_scaled.safetensors"
       source: "hf"

yaml/model_architecture_features.yaml

@@ -77,6 +77,11 @@ z-image:
   - vae
   - pid
 
+moss-image:
+  enabled_chains:
+  - lora
+  - pid
+
 ming-image:
   enabled_chains:
   - lora

yaml/model_architectures.yaml

@@ -2,6 +2,7 @@ architecture_order:
   - "FLUX.3"
   - "Krea-2"
   - "Qwen-Image-2.1"
+  - "MOSS-Image"
   - "Ming-Image"
   - "SenseNova-U1.5"
   - "Mage-Flow"
@@ -66,6 +67,8 @@ architectures:
   "Z-Image":
     model_type: "z-image"
     controlnet_key: "Z-Image"
+  "MOSS-Image":
+    model_type: "moss-image"
   "Ming-Image":
     model_type: "ming-image"
   "Qwen-Image-2.1":

yaml/model_defaults.yaml

@@ -22,6 +22,11 @@ FLUX.3:
     cfg: 1.0
     sampler_name: "euler"
     scheduler: "simple"
+  "FLUX-3-Image-Klein-Base":
+    steps: 50
+    cfg: 3.5
+    sampler_name: "euler"
+    scheduler: "simple"
 
 Krea-2:
   _defaults:
@@ -153,6 +158,13 @@ Z-Image:
     steps: 9
     cfg: 1.0
 
+MOSS-Image:
+  _defaults:
+    steps: 50
+    cfg: 3.0
+    sampler_name: "euler"
+    scheduler: "simple"
+
 Ming-Image:
   _defaults:
     steps: 12

yaml/model_list.yaml

@@ -4,7 +4,7 @@ Checkpoint:
     models:
       - display_name: "FLUX.3-Image-Klein"
         components:
-          unet: "flux3_image_klein_nvfp4.safetensors"
+          unet: "flux3-image-klein-nvfp4.safetensors"
           clip: "qwen3vl_4b_fp8_scaled.safetensors"
           vae: "flux2-vae.safetensors"
       - display_name: "FLUX.3-Image-Klein-Distilled"
@@ -12,6 +12,11 @@ Checkpoint:
           unet: "flux3_image_klein_sd_nvfp4.safetensors"
           clip: "qwen3vl_4b_fp8_scaled.safetensors"
           vae: "flux2-vae.safetensors"
+      - display_name: "FLUX-3-Image-Klein-Base"
+        components:
+          unet: "flux3-image-klein-base-nvfp4.safetensors"
+          clip: "qwen3vl_4b_fp8_scaled.safetensors"
+          vae: "flux2-vae.safetensors"
   Krea-2:
     latent_type: latent
     models:
@@ -58,6 +63,19 @@ Checkpoint:
           unet: "NoctQA_V1_int8_convrot.safetensors"
           vae: "qwen_image_2.1_vae_bf16.safetensors"
           clip: "qwen3vl_8b_nvfp4.safetensors"
+  MOSS-Image:
+    latent_type: flux2_latent
+    models:
+      - display_name: "MOSS-Image-SFT"
+        components:
+          unet: "moss_image_sft_int8_convrot.safetensors"
+          clip: "qwen3.5_9b_int8_convrot.safetensors"
+          vae: "flux2-vae.safetensors"
+      - display_name: "MOSS-Image-Base"
+        components:
+          unet: "moss_image_base_int8_convrot.safetensors"
+          clip: "qwen3.5_9b_int8_convrot.safetensors"
+          vae: "flux2-vae.safetensors"
   Ming-Image:
     latent_type: latent
     models:

yaml/pid.yaml

@@ -1,7 +1,7 @@
 PiD:
   - filepath: "pid_1.5_flux2_1024_to_4096_4step_int8_convrot.safetensors"
     latent_format: "flux"
-    architectures: ["flux3", "flux2", "flux2-kv", "ideogram-4", "lens", "ernie-image"]
+    architectures: ["flux3", "flux2", "flux2-kv", "ideogram-4", "lens", "ernie-image", "moss-image"]
   - filepath: "pid_1.5_qwenimage_1024_to_4096_4step_int8_convrot.safetensors"
     latent_format: "qwenimage"
     architectures: ["anima-3.8b-v1.1", "anima-3.8b", "anima", "qwen-image", "krea-2"]