Skip to content

GitLab

Commit 4ab59433 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Add config change to run script

parent b026de28
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
run_pretrained_openfold.py
View file @ 4ab59433
...@@ -44,7 +44,7 @@ from scripts.utils import add_data_args ...@@ -44,7 +44,7 @@ from scripts.utils import add_data_args
def main(args): def main(args):
config = model_config(args.model_name) config = model_config(args.model_name)
model = AlphaFold(config.model) model = AlphaFold(config)
model = model.eval() model = model.eval()
import_jax_weights_(model, args.param_path) import_jax_weights_(model, args.param_path)
model = model.to(args.model_device) model = model.to(args.model_device)
......
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