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

PiperOrigin-RevId: 495602713
Change-Id: I091644565b3bbb2e95ee434c6cd80d6d54d677dc

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",