Turn off bfloat16 in colab.

PiperOrigin-RevId: 514446997 Change-Id: I8b1113d39b62f7124d301e40c4b2e4418618bb7c

Turn off bfloat16 in colab.
PiperOrigin-RevId: 514446997 Change-Id: I8b1113d39b62f7124d301e40c4b2e4418618bb7c
651fe8cd · Hamish Tomlinson · Copybara-Service · 6918c085 · 651fe8cd
Commit 651fe8cd authored Mar 06, 2023 by Hamish Tomlinson Committed by Copybara-Service Mar 06, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

notebooks/AlphaFold.ipynb notebooks/AlphaFold.ipynb +1 -0

No files found.
--- a/notebooks/AlphaFold.ipynb
+++ b/notebooks/AlphaFold.ipynb
@@ -599,6 +599,7 @@
        "    pbar.set_description(f'Running {model_name}')\n",
        "\n",
        "    cfg = config.model_config(model_name)\n",
+        "    cfg.model.global_config.bfloat16 = False\n",
        "\n",
        "    if model_type_to_use == ModelType.MONOMER:\n",
        "      cfg.data.eval.num_ensemble = 1\n",