Restore the max sequence limit to 4000 now that unified memory is fixed.

PiperOrigin-RevId: 495602713 Change-Id: I091644565b3bbb2e95ee434c6cd80d6d54d677dc

Restore the max sequence limit to 4000 now that unified memory is fixed.
PiperOrigin-RevId: 495602713 Change-Id: I091644565b3bbb2e95ee434c6cd80d6d54d677dc
e80e2524 · Andrew Cowie · Copybara-Service · ed3ecabf · e80e2524
Commit e80e2524 authored Dec 15, 2022 by Andrew Cowie Committed by Copybara-Service Dec 15, 2022
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

notebooks/AlphaFold.ipynb notebooks/AlphaFold.ipynb +3 -3

No files found.
--- a/notebooks/AlphaFold.ipynb
+++ b/notebooks/AlphaFold.ipynb
@@ -262,9 +262,9 @@
        "    sequence_16, sequence_17, sequence_18, sequence_19, sequence_20)\n",
        "\n",
        "MIN_PER_SEQUENCE_LENGTH = 16\n",
-        "MAX_PER_SEQUENCE_LENGTH = 3400\n",
+        "MAX_PER_SEQUENCE_LENGTH = 4000\n",
        "MAX_MONOMER_MODEL_LENGTH = 2500\n",
-        "MAX_LENGTH = 3400\n",
+        "MAX_LENGTH = 4000\n",
        "MAX_VALIDATED_LENGTH = 3000\n",
        "\n",
        "#@markdown Select this checkbox to run the multimer model for a single sequence.\n",
@@ -274,7 +274,7 @@
        "#@markdown \n",
        "#@markdown \n",
        "#@markdown Due to improved memory efficiency the multimer model has a maximum\n",
-        "#@markdown limit of 3400 residues, while the monomer model has a limit of 2500\n",
+        "#@markdown limit of 4000 residues, while the monomer model has a limit of 2500\n",
        "#@markdown residues.\n",
        "\n",
        "use_multimer_model_for_monomers = False #@param {type:\"boolean\"}\n",