From 9bbe9671fb356dceb4590a3a62672056e6bfcb2b Mon Sep 17 00:00:00 2001 From: Tai An Date: Thu, 30 Apr 2026 00:25:32 -0700 Subject: [PATCH] fix(controlnet_hunyuan): use config.num_layers and forward pooled/style cond fields in from_transformer --- src/diffusers/models/controlnets/controlnet_hunyuan.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/diffusers/models/controlnets/controlnet_hunyuan.py b/src/diffusers/models/controlnets/controlnet_hunyuan.py index 6ef92d78dd6e..d31f3cb11cf2 100644 --- a/src/diffusers/models/controlnets/controlnet_hunyuan.py +++ b/src/diffusers/models/controlnets/controlnet_hunyuan.py @@ -189,7 +189,9 @@ def from_transformer( text_len_t5 = config.text_len_t5 conditioning_channels = conditioning_channels - transformer_num_layers = transformer_num_layers or config.transformer_num_layers + transformer_num_layers = transformer_num_layers or config.num_layers + pooled_projection_dim = config.pooled_projection_dim + use_style_cond_and_image_meta_size = config.use_style_cond_and_image_meta_size controlnet = cls( conditioning_channels=conditioning_channels, @@ -203,9 +205,11 @@ def from_transformer( mlp_ratio=mlp_ratio, num_attention_heads=num_attention_heads, patch_size=patch_size, + pooled_projection_dim=pooled_projection_dim, sample_size=sample_size, text_len=text_len, text_len_t5=text_len_t5, + use_style_cond_and_image_meta_size=use_style_cond_and_image_meta_size, ) if load_weights_from_transformer: key = controlnet.load_state_dict(transformer.state_dict(), strict=False)