double negative -> positive

double negative -> positive
This commit is contained in:
John Slegers 2024-08-13 14:50:35 +02:00
parent dd156556ca
commit ca94622f3f

View File

@ -238,7 +238,7 @@ class ControlNet(nn.Module):
else: else:
disabled_sa = False disabled_sa = False
if num_attention_blocks is not None or nr < num_attention_blocks[level]: if num_attention_blocks is None or nr < num_attention_blocks[level]:
layers.append( layers.append(
SpatialTransformer( SpatialTransformer(
ch, num_heads, dim_head, depth=num_transformers, context_dim=context_dim, ch, num_heads, dim_head, depth=num_transformers, context_dim=context_dim,