condition strategy

broadinstitute · Dec 16, 2024 · 46c5298 · 46c5298
1 parent e8e5228
commit 46c5298
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ml4h/models/diffusion_blocks.py b/ml4h/models/diffusion_blocks.py
@@ -212,7 +212,7 @@ def get_control_network(input_shape, widths, block_depth, kernel_size, control_s
         x = residual_block_control(widths[-1], conv, kernel_size, attention_heads, condition_strategy)([x, c2])
 
     for i, width in enumerate(reversed(widths[:-1])):
-        if attention_modulo > 1 and i % attention_modulo == 0:
+        if False and attention_modulo > 1 and i % attention_modulo == 0:
             if len(input_shape) > 2:
                 c2 = upsample(size=x.shape[1:-1])(control[control_idxs])
             else: