File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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
11671167def 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
You can’t perform that action at this time.
0 commit comments