Unverified Commit a6c9015c authored by laksjdjf's avatar laksjdjf Committed by GitHub
Browse files

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: default avatarSayak Paul <spsayakpaul@gmail.com>
parent e6a5f99e
...@@ -509,6 +509,9 @@ class ControlNetModel(ModelMixin, ConfigMixin, FromOriginalControlNetMixin): ...@@ -509,6 +509,9 @@ class ControlNetModel(ModelMixin, ConfigMixin, FromOriginalControlNetMixin):
if controlnet.class_embedding: if controlnet.class_embedding:
controlnet.class_embedding.load_state_dict(unet.class_embedding.state_dict()) 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.down_blocks.load_state_dict(unet.down_blocks.state_dict())
controlnet.mid_block.load_state_dict(unet.mid_block.state_dict()) controlnet.mid_block.load_state_dict(unet.mid_block.state_dict())
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment