Remove unnecessary double-load of config_json.

8b5212da · Ryan Kosai · f37d0d96 · 8b5212da
Commit 8b5212da authored Jul 16, 2024 by Ryan Kosai
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 6 deletions

run_pretrained_openfold.py run_pretrained_openfold.py +1 -6

No files found.
--- 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: 
+    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: 
        with open(args.experiment_config_json, 'r') as f:
            custom_config_dict = json.load(f)
        config.update_from_flattened_dict(custom_config_dict)