Improved UX: Automatically set the single seq mode flag

2023-10-05 13:16:22 -05:00 · 2023-10-05 13:16:22 -05:00 · e8de822e9b
parent c8c1239723
commit e8de822e9b
1 changed files with 2 additions and 0 deletions
--- a/run_pretrained_openfold.py
+++ b/run_pretrained_openfold.py
@ -152,6 +152,8 @@ def main(args):
    # Create the output directory
    os.makedirs(args.output_dir, exist_ok=True)

+    if args.config_preset.startswith("seq"):
+        args.use_single_seq_mode = True
    config = model_config(args.config_preset, long_sequence_inference=args.long_sequence_inference)

    if(args.trace_model):