Skip to content

Commit fdebe18

Browse files
Fix regular chroma radiance (Comfy-Org#11276)
1 parent f8321eb commit fdebe18

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

comfy/model_detection.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -261,6 +261,8 @@ def detect_unet_config(state_dict, key_prefix, metadata=None):
261261
dit_config["nerf_embedder_dtype"] = torch.float32
262262
if "__x0__" in state_dict_keys: # x0 pred
263263
dit_config["use_x0"] = True
264+
else:
265+
dit_config["use_x0"] = False
264266
else:
265267
dit_config["guidance_embed"] = "{}guidance_in.in_layer.weight".format(key_prefix) in state_dict_keys
266268
dit_config["yak_mlp"] = '{}double_blocks.0.img_mlp.gate_proj.weight'.format(key_prefix) in state_dict_keys

0 commit comments

Comments
 (0)