From 46c5298f0b74de34214b79b08bd51af7d2d075a3 Mon Sep 17 00:00:00 2001 From: Sam Freesun Friedman Date: Mon, 16 Dec 2024 12:19:04 -0500 Subject: [PATCH] condition strategy --- ml4h/models/diffusion_blocks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ml4h/models/diffusion_blocks.py b/ml4h/models/diffusion_blocks.py index 5f6855876..2fedf73cc 100644 --- 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: