From 3beef1b0b11efcd57281df1398604417d398b9e7 Mon Sep 17 00:00:00 2001 From: Sam Freesun Friedman Date: Tue, 10 Dec 2024 20:08:02 -0500 Subject: [PATCH] fix --- ml4h/models/diffusion_blocks.py | 2 +- ml4h/models/train.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ml4h/models/diffusion_blocks.py b/ml4h/models/diffusion_blocks.py index c974def66..10c0c4a22 100644 --- a/ml4h/models/diffusion_blocks.py +++ b/ml4h/models/diffusion_blocks.py @@ -209,7 +209,7 @@ def get_control_embed_model(output_maps, control_size): for cm in output_maps: control_ins.append(keras.Input(shape=cm.shape, name=cm.output_name())) c = layers.Concatenate()(control_ins) - c = layers.Dense(control_size, activation='linear')(c) + #c = layers.Dense(control_size, activation='linear')(c) return keras.Model(control_ins, c, name='control_embed') diff --git a/ml4h/models/train.py b/ml4h/models/train.py index ae979fe80..b6c9babc7 100755 --- a/ml4h/models/train.py +++ b/ml4h/models/train.py @@ -367,6 +367,6 @@ def train_diffusion_control_model(args): args.model_file = model_file eval_model, _, _, _ = make_multimodal_multitask_model(**args.__dict__) regress_on_controlled_generations(model, eval_model, tm_out, args.test_steps, args.batch_size, - 0.0,2.5,f'{args.output_folder}/{args.id}/') + 0.0,2.0,f'{args.output_folder}/{args.id}/') return model