mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-06-03 10:02:09 +08:00
Return proper error if diffusion model not detected properly. (#8272)
This commit is contained in:
parent
ad3bd8aa49
commit
a0651359d7
@ -620,6 +620,9 @@ def convert_config(unet_config):
|
|||||||
|
|
||||||
|
|
||||||
def unet_config_from_diffusers_unet(state_dict, dtype=None):
|
def unet_config_from_diffusers_unet(state_dict, dtype=None):
|
||||||
|
if "conv_in.weight" not in state_dict:
|
||||||
|
return None
|
||||||
|
|
||||||
match = {}
|
match = {}
|
||||||
transformer_depth = []
|
transformer_depth = []
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user