Add ldm format support to UNETLoader.

comfy/sd.py

@@ -454,7 +454,13 @@ def load_unet(unet_path): #load unet in diffusers format
     sd = comfy.utils.load_torch_file(unet_path)
     parameters = comfy.utils.calculate_parameters(sd)
     fp16 = model_management.should_use_fp16(model_params=parameters)
+    if "input_blocks.0.0.weight" in sd: #ldm
+        model_config = model_detection.model_config_from_unet(sd, "", fp16)
+        if model_config is None:
+            raise RuntimeError("ERROR: Could not detect model type of: {}".format(unet_path))
+        new_sd = sd
 
+    else: #diffusers
         model_config = model_detection.model_config_from_diffusers_unet(sd, fp16)
         if model_config is None:
             print("ERROR UNSUPPORTED UNET", unet_path)