Skip to content

Commit b327a26

Browse files
committed
transformer_layers -> transformer_layes
1 parent c5cb0b1 commit b327a26

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/models/controlnet_union.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -366,7 +366,7 @@ def __init__(
366366

367367
task_scale_factor = num_trans_channel**0.5
368368
self.task_embedding = nn.Parameter(task_scale_factor * torch.randn(num_control_type, num_trans_channel))
369-
self.transformer_layers = nn.ModuleList(
369+
self.transformer_layes = nn.ModuleList(
370370
[ResidualAttentionBlock(num_trans_channel, num_trans_head) for _ in range(num_trans_layer)]
371371
)
372372
self.spatial_ch_projs = zero_module(nn.Linear(num_trans_channel, num_proj_channel))
@@ -837,7 +837,7 @@ def forward(
837837
condition_list.append(condition)
838838

839839
x = torch.cat(inputs, dim=1)
840-
for layer in self.transformer_layers:
840+
for layer in self.transformer_layes:
841841
x = layer(x)
842842

843843
controlnet_cond_fuser = sample * 0.0

0 commit comments

Comments
 (0)