Fix bug in inference script weight importing

341ff3d9 · Gustaf Ahdritz · GitHub · 6448d57c · 341ff3d9
Unverified Commit 341ff3d9 authored Nov 25, 2021 by Gustaf Ahdritz Committed by GitHub Nov 25, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

run_pretrained_openfold.py run_pretrained_openfold.py +1 -1

No files found.
--- a/run_pretrained_openfold.py
+++ b/run_pretrained_openfold.py
@@ -48,7 +48,7 @@ def main(args):
    config = model_config(args.model_name)
    model = AlphaFold(config)
    model = model.eval()
-    import_jax_weights_(model, args.param_path)
+    import_jax_weights_(model, args.param_path, version=args.model_name)
    #script_preset_(model)
    model = model.to(args.model_device)