Add FLUX.2 architecture model

core/pipelines/workflow_recipes/_partials/conditioning/flux2.yaml

@@ -0,0 +1,51 @@
+nodes:
+  unet_loader:
+    class_type: UNETLoader
+    title: "Load Diffusion Model"
+    params:
+      weight_dtype: "default"
+  clip_loader:
+    class_type: CLIPLoader
+    title: "Load CLIP"
+    params:
+      type: "flux2"
+      device: "default"
+  vae_loader:
+    class_type: VAELoader
+    title: "Load VAE"
+  
+  pos_prompt:
+    class_type: CLIPTextEncode
+    title: "CLIP Text Encode (Positive)"
+  neg_prompt:
+    class_type: CLIPTextEncode
+    title: "CLIP Text Encode (Negative)"
+
+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: "vae_loader:0"
+    to: "vae_decode:vae"
+  - from: "vae_loader:0"
+    to: "vae_encode:vae"
+  
+  - from: "pos_prompt:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    ksampler_node: "ksampler"
+    clip_source: "clip_loader:0"
+
+ui_map:
+  unet_name: "unet_loader:unet_name"
+  clip_name: "clip_loader:clip_name"
+  vae_name: "vae_loader:vae_name"
+  positive_prompt: "pos_prompt:text"
+  negative_prompt: "neg_prompt:text"

yaml/file_list.yaml

@@ -8,3 +8,18 @@ file:
       source: "hf"
       repo_id: "stable-diffusion-v1-5/stable-diffusion-v1-5"
       repository_file_path: "v1-5-pruned-emaonly.safetensors"
+  diffusion_models:
+    - filename: "flux2_dev_fp8mixed.safetensors"
+      source: "hf"
+      repo_id: "Comfy-Org/flux2-dev"
+      repository_file_path: "split_files/diffusion_models/flux2_dev_fp8mixed.safetensors"
+  text_encoders:
+    - filename: "mistral_3_small_flux2_fp8.safetensors"
+      source: "hf"
+      repo_id: "Comfy-Org/flux2-dev"
+      repository_file_path: "split_files/text_encoders/mistral_3_small_flux2_fp8.safetensors"
+  vae:
+    - filename: "flux2-vae.safetensors"
+      source: "hf"
+      repo_id: "Comfy-Org/flux2-dev"
+      repository_file_path: "split_files/vae/flux2-vae.safetensors"

yaml/model_architectures.yaml

@@ -1,8 +1,12 @@
 architecture_order:
+  - "FLUX.2"
   - "SDXL"
   - "SD1.5"
 
 architectures:
+  "FLUX.2":
+    model_type: "flux2"
+    controlnet_key: "FLUX.2"
   "SDXL":
     model_type: "sdxl"
     controlnet_key: "SDXL"

yaml/model_defaults.yaml

@@ -1,4 +1,11 @@
 defaults:
+  FLUX.2:
+    width: 1024
+    height: 1024
+    num_inference_steps: 28
+    guidance_scale: 3.5
+    sampler: "Euler"
+    schedule_type: "Default"
   SDXL:
     width: 1024
     height: 1024

yaml/model_list.yaml

@@ -1,4 +1,12 @@
 Checkpoint:
+  FLUX.2:
+    latent_type: flux2_latent
+    models:
+      - display_name: "Comfy-Org/flux2-dev"
+        components:
+          unet: "flux2_dev_fp8mixed.safetensors"
+          clip: "mistral_3_small_flux2_fp8.safetensors"
+          vae: "flux2-vae.safetensors"
   SDXL:
     latent_type: latent
     models:

yaml/private_file_list.yaml

@@ -0,0 +1,12 @@
+file:
+  diffusion_models: 
+    # FLUX.2-klein-9B
+    - filename: "flux-2-klein-9b-fp8.safetensors"
+      source: "hf"
+      repo_id: "black-forest-labs/FLUX.2-klein-9b-fp8"
+      repository_file_path: "flux-2-klein-9b-fp8.safetensors"
+    # FLUX.2-klein-base-9B
+    - filename: "flux-2-klein-base-9b-fp8.safetensors"
+      source: "hf"
+      repo_id: "black-forest-labs/FLUX.2-klein-base-9b-fp8"
+      repository_file_path: "flux-2-klein-base-9b-fp8.safetensors"