Fix notebook again

e6852aee · Gustaf Ahdritz · 892adf48 · e6852aee
Commit e6852aee authored Jul 15, 2022 by Gustaf Ahdritz
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

notebooks/OpenFold.ipynb notebooks/OpenFold.ipynb +2 -0

No files found.
--- a/notebooks/OpenFold.ipynb
+++ b/notebooks/OpenFold.ipynb
@@ -467,6 +467,8 @@
        "      config_preset = f\"model_{i}\"\n",
        "    else:\n",
        "      config_preset = \"model_1\"\n",
+        "      if(\"_ptm_\" in model_name):\n",
+        "        config_preset += \"_ptm\"\n",
        "\n",
        "    cfg = config.model_config(config_preset)\n",
        "    openfold_model = model.AlphaFold(cfg)\n",