From fdebe182966d1dd9bee3138264937137bd2302d8 Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Thu, 11 Dec 2025 14:09:35 -0800
Subject: [PATCH 1/2] Fix regular chroma radiance (#11276)

---
 comfy/model_detection.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/comfy/model_detection.py b/comfy/model_detection.py
index 1f5d34bddae4..94b54b7c2c0d 100644
--- a/comfy/model_detection.py
+++ b/comfy/model_detection.py
@@ -261,6 +261,8 @@ def detect_unet_config(state_dict, key_prefix, metadata=None):
                 dit_config["nerf_embedder_dtype"] = torch.float32
             if "__x0__" in state_dict_keys: # x0 pred
                 dit_config["use_x0"] = True
+            else:
+                dit_config["use_x0"] = False
         else:
             dit_config["guidance_embed"] = "{}guidance_in.in_layer.weight".format(key_prefix) in state_dict_keys
             dit_config["yak_mlp"] = '{}double_blocks.0.img_mlp.gate_proj.weight'.format(key_prefix) in state_dict_keys

From ae65433a602470eea271df47af0eb871d146a002 Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Thu, 11 Dec 2025 14:15:00 -0800
Subject: [PATCH 2/2] This only works on radiance. (#11277)

---
 comfy/model_detection.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/comfy/model_detection.py b/comfy/model_detection.py
index 94b54b7c2c0d..dd6a703f63c1 100644
--- a/comfy/model_detection.py
+++ b/comfy/model_detection.py
@@ -259,10 +259,10 @@ def detect_unet_config(state_dict, key_prefix, metadata=None):
                 dit_config["nerf_tile_size"] = 512
                 dit_config["nerf_final_head_type"] = "conv" if f"{key_prefix}nerf_final_layer_conv.norm.scale" in state_dict_keys else "linear"
                 dit_config["nerf_embedder_dtype"] = torch.float32
-            if "__x0__" in state_dict_keys: # x0 pred
-                dit_config["use_x0"] = True
-            else:
-                dit_config["use_x0"] = False
+                if "__x0__" in state_dict_keys: # x0 pred
+                    dit_config["use_x0"] = True
+                else:
+                    dit_config["use_x0"] = False
         else:
             dit_config["guidance_embed"] = "{}guidance_in.in_layer.weight".format(key_prefix) in state_dict_keys
             dit_config["yak_mlp"] = '{}double_blocks.0.img_mlp.gate_proj.weight'.format(key_prefix) in state_dict_keys