File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments