From a6c9015c4e0f608a5a45ce031087069f34a94a24 Mon Sep 17 00:00:00 2001 From: laksjdjf Date: Wed, 20 Mar 2024 04:45:08 +0900 Subject: [PATCH] Fix ControlNetModel.from_unet do not load add_embedding (#7269) * Fix ControlNetModel.from_unet do not load add_embedding * delete white space in blank line --------- Co-authored-by: Sayak Paul --- src/diffusers/models/controlnet.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/diffusers/models/controlnet.py b/src/diffusers/models/controlnet.py index ebd4133a6d..200881c0ed 100644 --- a/src/diffusers/models/controlnet.py +++ b/src/diffusers/models/controlnet.py @@ -509,6 +509,9 @@ class ControlNetModel(ModelMixin, ConfigMixin, FromOriginalControlNetMixin): if controlnet.class_embedding: controlnet.class_embedding.load_state_dict(unet.class_embedding.state_dict()) + if hasattr(controlnet, "add_embedding"): + controlnet.add_embedding.load_state_dict(unet.add_embedding.state_dict()) + controlnet.down_blocks.load_state_dict(unet.down_blocks.state_dict()) controlnet.mid_block.load_state_dict(unet.mid_block.state_dict())