We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e6a5f99 commit a6c9015Copy full SHA for a6c9015
src/diffusers/models/controlnet.py
@@ -509,6 +509,9 @@ def from_unet(
509
if controlnet.class_embedding:
510
controlnet.class_embedding.load_state_dict(unet.class_embedding.state_dict())
511
512
+ if hasattr(controlnet, "add_embedding"):
513
+ controlnet.add_embedding.load_state_dict(unet.add_embedding.state_dict())
514
+
515
controlnet.down_blocks.load_state_dict(unet.down_blocks.state_dict())
516
controlnet.mid_block.load_state_dict(unet.mid_block.state_dict())
517
0 commit comments