Improved UX: Automatically set the single seq mode flag

aqlaboratory · Oct 10, 2023 · e8de822 · e8de822
1 parent c8c1239
commit e8de822
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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):