From 2bb6fa78eeebd06fbaf87cbd8181a5fb5ba071af Mon Sep 17 00:00:00 2001 From: Adeel Ahmad Date: Tue, 21 Jul 2026 16:27:02 +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.