diff --git a/comfy/model_detection.py b/comfy/model_detection.py index 475b43d90..13fa1fa5e 100644 --- a/comfy/model_detection.py +++ b/comfy/model_detection.py @@ -258,7 +258,6 @@ def detect_unet_config(state_dict, key_prefix, metadata=None): 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 f"__x0__" in state_dict_keys: # x0 pred - print("radiance in x0 mode") dit_config["use_x0"] = True else: dit_config["guidance_embed"] = "{}guidance_in.in_layer.weight".format(key_prefix) in state_dict_keys