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:
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(
SpatialTransformer(
ch, num_heads, dim_head, depth=num_transformers, context_dim=context_dim,