full codespell coverage

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-06-04 12:36:10 +02:00
parent 0bfcdabbbb
commit 5e99dee3c2
103 changed files with 264 additions and 254 deletions
@@ -459,7 +459,7 @@ class split_AttnProcessor2_0(torch.nn.Module):
hidden_states_0 = hidden_states_0.view(batch_size, channel, height * width).transpose(1, 2)
hidden_states_1 = hidden_states_1.view(batch_size, channel, height * width).transpose(1, 2)
else:
# directly split sqeuence according to concat dim.
# directly split sequence according to concat dim.
single_dim = original_shape[2] if cat_dim==-2 or cat_dim==2 else original_shape[1]
hidden_states_0 = hidden_states[:, :single_dim*single_dim,:]
hidden_states_1 = hidden_states[:, single_dim*(single_dim+1):,:]
@@ -593,7 +593,7 @@ class sep_split_AttnProcessor2_0(torch.nn.Module):
hidden_states_0 = hidden_states_0.view(batch_size, channel, height * width).transpose(1, 2)
hidden_states_1 = hidden_states_1.view(batch_size, channel, height * width).transpose(1, 2)
else:
# directly split sqeuence according to concat dim.
# directly split sequence according to concat dim.
single_dim = original_shape[2] if cat_dim==-2 or cat_dim==2 else original_shape[1]
hidden_states_0 = hidden_states[:, :single_dim*single_dim,:]
hidden_states_1 = hidden_states[:, single_dim*(single_dim+1):,:]