From c67b06bdd6545a7b3c08f22d08522bbfba41515a Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Sun, 25 May 2025 03:58:41 -0400
Subject: [PATCH] Return proper error if diffusion model not detected properly.

---
 comfy/model_detection.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/comfy/model_detection.py b/comfy/model_detection.py
index 20f287df9..74f539598 100644
--- a/comfy/model_detection.py
+++ b/comfy/model_detection.py
@@ -620,6 +620,9 @@ def convert_config(unet_config):
 
 
 def unet_config_from_diffusers_unet(state_dict, dtype=None):
+    if "conv_in.weight" not in state_dict:
+        return None
+
     match = {}
     transformer_depth = []