Upload folder using huggingface_hub

README.md

@@ -43,6 +43,7 @@ models:
   - Comfy-Org/HiDream-O1-Image
   - Comfy-Org/HunyuanImage_2.1_ComfyUI
   - Comfy-Org/Ideogram-4
+  - Comfy-Org/Krea-2
   - Comfy-Org/Lens
   - Comfy-Org/LongCat-Image
   - Comfy-Org/Lumina_Image_2.0_Repackaged

core/pipelines/workflow_recipes/_partials/conditioning/krea-2.yaml

@@ -0,0 +1,58 @@
+nodes:
+  unet_loader:
+    class_type: UNETLoader
+    title: "Load Diffusion Model"
+    params:
+      weight_dtype: "default"
+  clip_loader:
+    class_type: CLIPLoader
+    title: "Load CLIP"
+    params:
+      type: "krea2"
+      device: "default"
+  vae_loader:
+    class_type: VAELoader
+    title: "Load VAE"
+
+connections:
+  - from: "unet_loader: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": ["ksampler:model"]
+      "clip": ["pos_prompt:clip", "neg_prompt:clip"]
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    ksampler_node: "ksampler"
+    clip_source: "clip_loader:0"
+
+dynamic_pid_chains:
+  pid_chain:
+    ksampler_node: "ksampler"
+
+ui_map:
+  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.45.19
-comfyui-workflow-templates==0.10.0
+comfyui-workflow-templates==0.10.3
 comfyui-embedded-docs==0.5.5
 torch
 torchsde

yaml/constants.yaml

@@ -7,6 +7,14 @@ MAX_REFERENCE_LATENTS: 10
 LORA_SOURCE_CHOICES: ["Civitai", "File"]
 
 RESOLUTION_MAP:
+  krea-2:
+    "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]
   boogu-image:
     "1:1 (Square)": [1024, 1024]
     "16:9 (Landscape)": [1344, 768]

yaml/file_list.yaml

@@ -403,6 +403,15 @@ file:
       repo_id: "comfyanonymous/ControlNet-v1-1_fp16_safetensors"
       repository_file_path: "control_v11u_sd15_tile_fp16.safetensors"
   diffusion_models: 
+    # Krea-2
+    - filename: "krea2_turbo_nvfp4.safetensors"
+      source: "hf"
+      repo_id: "Comfy-Org/Krea-2"
+      repository_file_path: "diffusion_models/krea2_turbo_nvfp4.safetensors"
+    - filename: "krea2_raw_fp8_scaled.safetensors"
+      source: "hf"
+      repo_id: "Comfy-Org/Krea-2"
+      repository_file_path: "diffusion_models/krea2_raw_fp8_scaled.safetensors"
     # Boogu-Image
     - filename: "boogu_image_base_nvfp4.safetensors"
       source: "hf"
@@ -729,6 +738,11 @@ file:
       repo_id: "alibaba-pai/Z-Image-Turbo-Fun-Controlnet-Union-2.1"
       repository_file_path: "Z-Image-Turbo-Fun-Controlnet-Tile-2.1-8steps.safetensors"
   text_encoders: 
+    # Krea-2
+    - filename: "qwen3vl_4b_fp8_scaled.safetensors"
+      source: "hf"
+      repo_id: "Comfy-Org/Krea-2"
+      repository_file_path: "text_encoders/qwen3vl_4b_fp8_scaled.safetensors"
     # Cosmos-Predict2
     - filename: "oldt5_xxl_fp8_e4m3fn_scaled.safetensors"
       source: "hf"

yaml/image_gen_features.yaml

@@ -8,6 +8,12 @@ default:
   - conditioning
   - vae
 
+krea-2:
+  enabled_chains:
+    - lora
+    - conditioning
+    - pid
+
 boogu-image:
   enabled_chains:
   - lora

yaml/model_architectures.yaml

@@ -1,4 +1,5 @@
 architecture_order:
+  - "Krea-2"
   - "Boogu-Image"
   - "PixelDiT"
   - "Ideogram-4"
@@ -27,21 +28,18 @@ architecture_order:
   - "SD1.5"
 
 architectures:
+  Krea-2:
+    model_type: "krea-2"
   "Boogu-Image":
     model_type: "boogu-image"
-    controlnet_key: "Boogu-Image"
   "PixelDiT":
     model_type: "pixeldit"
-    controlnet_key: "PixelDiT"
   "Ideogram-4":
     model_type: "ideogram-4"
-    controlnet_key: "Ideogram-4"
   "Lens":
     model_type: "lens"
-    controlnet_key: "Lens"
   "ERNIE-Image":
     model_type: "ernie-image"
-    controlnet_key: "ERNIE-Image"
   "FLUX.2-KV":
     model_type: "flux2-kv"
     controlnet_key: "FLUX.2"

yaml/model_defaults.yaml

@@ -1,11 +1,23 @@
 Default:
-  steps: 4
+  steps: 8
   cfg: 1.0
-  sampler_name: "lcm"
-  scheduler: "sgm_uniform"
+  sampler_name: "euler"
+  scheduler: "simple"
   positive_prompt: ""
   negative_prompt: ""
 
+Krea-2:
+  _defaults:
+    steps: 52
+    cfg: 3.5
+    sampler_name: "euler"
+    scheduler: "simple"
+  "Krea-2-Turbo":
+    steps: 8
+    cfg: 1.0
+    sampler_name: "euler"
+    scheduler: "simple"
+
 Boogu-Image:
   _defaults:
     steps: 25

yaml/model_list.yaml

@@ -1,4 +1,17 @@
 Checkpoint:
+  Krea-2:
+    latent_type: latent
+    models:
+      - display_name: "Krea-2-Turbo"
+        components:
+          unet: "krea2_turbo_nvfp4.safetensors"
+          clip: "qwen3vl_4b_fp8_scaled.safetensors"
+          vae: "qwen_image_vae.safetensors"
+      - display_name: "Krea-2-Raw"
+        components:
+          unet: "krea2_raw_fp8_scaled.safetensors"
+          clip: "qwen3vl_4b_fp8_scaled.safetensors"
+          vae: "qwen_image_vae.safetensors"
   Boogu-Image:
     latent_type: latent
     models:
@@ -12,7 +25,6 @@ Checkpoint:
           unet: "boogu_image_base_nvfp4.safetensors"
           clip: "qwen3vl_8b_nvfp4.safetensors"
           vae: "ae.safetensors"
-
   PixelDiT:
     latent_type: chroma_radiance_latent
     models:

yaml/pid.yaml

@@ -4,7 +4,7 @@ PiD:
     architectures: ["flux2", "flux2-kv", "ideogram-4", "lens", "ernie-image"]
   - filepath: "pid_qwenimage_1024_to_4096_4step_bf16.safetensors"
     latent_format: "qwenimage"
-    architectures: ["anima", "qwen-image"]
+    architectures: ["anima", "qwen-image", "krea-2"]
   - filepath: "pid_flux1_1024_to_4096_4step_mxfp8.safetensors"
     latent_format: "flux"
     architectures: ["longcat-image", "newbie-image", "z-image", "kandinsky-5", "ovis-image", "omnigen2", "chroma1", "hidream-i1", "flux1", "boogu-image"]