From 7a13f74220dcfcf6d8a6e7c1984f2c718cd9eda8 Mon Sep 17 00:00:00 2001 From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com> Date: Wed, 25 Jun 2025 01:52:34 -0700 Subject: [PATCH] unet -> diffusion model (#8659) --- comfy/sd.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/comfy/sd.py b/comfy/sd.py index cd13ab5f..c32cf6b6 100644 --- a/comfy/sd.py +++ b/comfy/sd.py @@ -1160,7 +1160,7 @@ def load_diffusion_model_state_dict(sd, model_options={}): model.load_model_weights(new_sd, "") left_over = sd.keys() if len(left_over) > 0: - logging.info("left over keys in unet: {}".format(left_over)) + logging.info("left over keys in diffusion model: {}".format(left_over)) return comfy.model_patcher.ModelPatcher(model, load_device=load_device, offload_device=offload_device) @@ -1168,7 +1168,7 @@ def load_diffusion_model(unet_path, model_options={}): sd = comfy.utils.load_torch_file(unet_path) model = load_diffusion_model_state_dict(sd, model_options=model_options) if model is None: - logging.error("ERROR UNSUPPORTED UNET {}".format(unet_path)) + logging.error("ERROR UNSUPPORTED DIFFUSION MODEL {}".format(unet_path)) raise RuntimeError("ERROR: Could not detect model type of: {}".format(unet_path)) return model