Update src/diffusers/models/controlnet_sd3.py

Co-authored-by: YiYi Xu <yixu310@gmail.com>
haofanwang · Jul 1, 2024 · 8ad1ca5 · 8ad1ca5
1 parent 33791cf
commit 8ad1ca5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/diffusers/models/controlnet_sd3.py b/src/diffusers/models/controlnet_sd3.py
@@ -246,7 +246,7 @@ def from_transformer(cls, transformer, num_layers=12, load_weights_from_transfor
 
         if load_weights_from_transformer:
             controlnet.pos_embed.load_state_dict(transformer.pos_embed.state_dict(), strict=False)
-            controlnet.time_text_embed.load_state_dict(transformer.time_text_embed.state_dict(), strict=False)
+            controlnet.time_text_embed.load_state_dict(transformer.time_text_embed.state_dict())
             controlnet.context_embedder.load_state_dict(transformer.context_embedder.state_dict(), strict=False)
             controlnet.transformer_blocks.load_state_dict(transformer.transformer_blocks.state_dict(), strict=False)