diff --git a/run_pretrained_openfold.py b/run_pretrained_openfold.py index eb4ad7c0..e4fd6ad4 100644 --- a/run_pretrained_openfold.py +++ b/run_pretrained_openfold.py @@ -185,12 +185,7 @@ def main(args): use_deepspeed_evoformer_attention=args.use_deepspeed_evoformer_attention, ) - if args.experiment_config_json: - with open(args.experiment_config_json, 'r') as f: - custom_config_dict = json.load(f) - config.update_from_flattened_dict(custom_config_dict) - - if args.experiment_config_json: + if args.experiment_config_json: with open(args.experiment_config_json, 'r') as f: custom_config_dict = json.load(f) config.update_from_flattened_dict(custom_config_dict)