Skip to content

GitLab

Commit e8de822e authored by Sachin Kadyan's avatar Sachin Kadyan
Browse files

Improved UX: Automatically set the single seq mode flag

parent c8c12397
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
run_pretrained_openfold.py
View file @ e8de822e
...@@ -152,6 +152,8 @@ def main(args): ...@@ -152,6 +152,8 @@ def main(args):
# Create the output directory # Create the output directory
os.makedirs(args.output_dir, exist_ok=True) 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) config = model_config(args.config_preset, long_sequence_inference=args.long_sequence_inference)
if(args.trace_model): if(args.trace_model):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment