Remove unused tflite libraries that cause a crash in colab

Author: milot-mirdita

AlphaFold2.ipynb

@@ -151,7 +151,7 @@
         "  os.system(\"ln -s /usr/local/lib/python3.*/dist-packages/colabfold colabfold\")\n",
         "  os.system(\"ln -s /usr/local/lib/python3.*/dist-packages/alphafold alphafold\")\n",
         "  # hack to fix TF crash\n",
-        "  os.system(\"rm -f /usr/local/lib/python3.*/dist-packages/tensorflow/core/kernels/libtfkernel_sobol_op.so\")\n",
+        "  os.system(\"rm -f /usr/local/lib/python3.*/dist-packages/tensorflow/core/kernels/libtfkernel_sobol_op.so /usr/local/lib/python3.*/dist-packages/tensorflow/lite/python/*/*.so\")\n",
         "  os.system(\"touch COLABFOLD_READY\")\n",
         "\n",
         "if USE_AMBER or USE_TEMPLATES:\n",

Boltz1.ipynb

@@ -249,7 +249,7 @@
         "  os.system(\"ln -s /usr/local/lib/python3.*/dist-packages/colabfold colabfold\")\n",
         "  os.system(\"ln -s /usr/local/lib/python3.*/dist-packages/alphafold alphafold\")\n",
         "  # hack to fix TF crash\n",
-        "  os.system(\"rm -f /usr/local/lib/python3.*/dist-packages/tensorflow/core/kernels/libtfkernel_sobol_op.so\")\n",
+        "  os.system(\"rm -f /usr/local/lib/python3.*/dist-packages/tensorflow/core/kernels/libtfkernel_sobol_op.so /usr/local/lib/python3.*/dist-packages/tensorflow/lite/python/*/*.so\")\n",
         "  os.system(\"touch COLABFOLD_READY\")\n",
         "\n",
         "if not os.path.isfile(\"BOLZ_READY\"):\n",

batch/AlphaFold2_batch.ipynb

@@ -136,7 +136,7 @@
         "  ln -s /usr/local/lib/python3.*/dist-packages/colabfold colabfold\n",
         "  ln -s /usr/local/lib/python3.*/dist-packages/alphafold alphafold\n",
         "  # hack to fix TF crash\n",
-        "  rm -f /usr/local/lib/python3.*/dist-packages/tensorflow/core/kernels/libtfkernel_sobol_op.so\n",
+        "  rm -f /usr/local/lib/python3.*/dist-packages/tensorflow/core/kernels/libtfkernel_sobol_op.so /usr/local/lib/python3.*/dist-packages/tensorflow/lite/python/*/*.so\n",
         "  touch COLABFOLD_READY\n",
         "fi\n",
         "\n",