Can't load the model files. The same error whether it is 4 or 8.
Error occurred when executing CheckpointLoaderSimple:
'model.diffusion_model.input_blocks.0.0.weight'
File "/home/developer/ComfyUI/execution.py", line 152, in recursive_execute
output_data, output_ui = get_output_data(obj, input_data_all)
File "/home/developer/ComfyUI/execution.py", line 82, in get_output_data
return_values = map_node_over_list(obj, input_data_all, obj.FUNCTION, allow_interrupt=True)
File "/home/developer/ComfyUI/execution.py", line 75, in map_node_over_list
results.append(getattr(obj, func)(**slice_dict(input_data_all, i)))
File "/home/developer/ComfyUI/nodes.py", line 540, in load_checkpoint
out = comfy.sd.load_checkpoint_guess_config(ckpt_path, output_vae=True, output_clip=True, embedding_directory=folder_paths.get_folder_paths("embeddings"))
File "/home/developer/ComfyUI/comfy/sd.py", line 506, in load_checkpoint_guess_config
model_config = model_detection.model_config_from_unet(sd, "model.diffusion_model.")
File "/home/developer/ComfyUI/comfy/model_detection.py", line 193, in model_config_from_unet
unet_config = detect_unet_config(state_dict, unet_key_prefix)
File "/home/developer/ComfyUI/comfy/model_detection.py", line 77, in detect_unet_config
model_channels = state_dict['{}input_blocks.0.0.weight'.format(key_prefix)].shape[0]
You probably downloaded the wrong weight.
Usesdxl_lightning_4step.safetensors
Notsdxl_lightning_4step_lora.safetensors
sdxl_lightning_4step_unet.safetensors