Skip to content

Commit a6c9015

Browse files
laksjdjfsayakpaul
andauthored
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 <[email protected]>
1 parent e6a5f99 commit a6c9015

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/diffusers/models/controlnet.py

+3
Original file line numberDiff line numberDiff line change
@@ -509,6 +509,9 @@ def from_unet(
509509
if controlnet.class_embedding:
510510
controlnet.class_embedding.load_state_dict(unet.class_embedding.state_dict())
511511

512+
if hasattr(controlnet, "add_embedding"):
513+
controlnet.add_embedding.load_state_dict(unet.add_embedding.state_dict())
514+
512515
controlnet.down_blocks.load_state_dict(unet.down_blocks.state_dict())
513516
controlnet.mid_block.load_state_dict(unet.mid_block.state_dict())
514517

0 commit comments

Comments
 (0)