From fb1363854350635f3685fb8d370d694c52be2b9d Mon Sep 17 00:00:00 2001 From: Adeel Ahmad Date: Wed, 22 Jul 2026 15:46:35 +0100 Subject: [PATCH] Clarify DualTransformer2DModel forward in docstring --- src/diffusers/models/transformers/dual_transformer_2d.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/diffusers/models/transformers/dual_transformer_2d.py b/src/diffusers/models/transformers/dual_transformer_2d.py index 778d5128ee23..b227c057b963 100644 --- a/src/diffusers/models/transformers/dual_transformer_2d.py +++ b/src/diffusers/models/transformers/dual_transformer_2d.py @@ -21,6 +21,8 @@ class DualTransformer2DModel(nn.Module): """ Dual transformer wrapper that combines two `Transformer2DModel`s for mixed inference. + Both sub-transformers are invoked directly inside `forward`. + Parameters: num_attention_heads (`int`, *optional*, defaults to 16): The number of heads to use for multi-head attention. attention_head_dim (`int`, *optional*, defaults to 88): The number of channels in each head.