Skip to content

Commit 7a13f74

Browse files
unet -> diffusion model (Comfy-Org#8659)
1 parent 8042eb2 commit 7a13f74

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

comfy/sd.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1160,15 +1160,15 @@ def load_diffusion_model_state_dict(sd, model_options={}):
11601160
model.load_model_weights(new_sd, "")
11611161
left_over = sd.keys()
11621162
if len(left_over) > 0:
1163-
logging.info("left over keys in unet: {}".format(left_over))
1163+
logging.info("left over keys in diffusion model: {}".format(left_over))
11641164
return comfy.model_patcher.ModelPatcher(model, load_device=load_device, offload_device=offload_device)
11651165

11661166

11671167
def load_diffusion_model(unet_path, model_options={}):
11681168
sd = comfy.utils.load_torch_file(unet_path)
11691169
model = load_diffusion_model_state_dict(sd, model_options=model_options)
11701170
if model is None:
1171-
logging.error("ERROR UNSUPPORTED UNET {}".format(unet_path))
1171+
logging.error("ERROR UNSUPPORTED DIFFUSION MODEL {}".format(unet_path))
11721172
raise RuntimeError("ERROR: Could not detect model type of: {}".format(unet_path))
11731173
return model
11741174

0 commit comments

Comments
 (0)