wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 14:50:49 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Ensure unet is defined

Browse Source
This commit is contained in:
Max Tretikov 2024-06-13 23:25:36 -06:00
parent 61b484d546
commit f9c3779d4e
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/diffusers_load.py
View File

@ -26,6 +26,7 @@ def load_diffusers(model_path, output_vae=True, output_clip=True, embedding_dire
if text_encoder2_path is not None:
text_encoder_paths.append(text_encoder2_path)
unet = None
if unet_path is not None:
unet = sd.load_unet(unet_path)