Update config reference in notebook

98d42cf7 · Gustaf Ahdritz · c6554a0b · 98d42cf7
Commit 98d42cf7 authored Nov 12, 2021 by Gustaf Ahdritz
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

notebooks/OpenFold.ipynb notebooks/OpenFold.ipynb +1 -1

No files found.
--- a/notebooks/OpenFold.ipynb
+++ b/notebooks/OpenFold.ipynb
@@ -426,7 +426,7 @@
        "    feature_dict.update(_placeholder_template_feats(num_templates, num_res))\n",
        "\n",
        "    cfg = config.model_config(model_name)\n",
-        "    openfold_model = model.AlphaFold(cfg.model)\n",
+        "    openfold_model = model.AlphaFold(cfg)\n",
        "    openfold_model = openfold_model.eval()\n",
        "    params_name = os.path.join(PARAMS_DIR, f\"params_{model_name}.npz\")\n",
        "    import_jax_weights_(openfold_model, params_name, version=model_name)\n",