fix esmfold params

sokrypton · Aug 26, 2023 · 120e031 · 120e031
1 parent a382230
commit 120e031
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/ESMFold.ipynb b/ESMFold.ipynb
@@ -60,14 +60,15 @@
         "#@title install\n",
         "#@markdown install ESMFold, OpenFold and download Params (~2min 30s)\n",
         "version = \"1\" # @param [\"0\", \"1\"]\n",
-        "model_name = \"esmfold_v0.model\" if version == \"0\" else \"esmfold_v0.model\"\n",
+        "model_name = \"esmfold_v0.model\" if version == \"0\" else \"esmfold.model\"\n",
         "import os, time\n",
         "if not os.path.isfile(model_name):\n",
         "  # download esmfold params\n",
         "  os.system(\"apt-get install aria2 -qq\")\n",
         "  os.system(f\"aria2c -q -x 16 https://colabfold.steineggerlab.workers.dev/esm/{model_name} &\")\n",
         "\n",
         "  if not os.path.isfile(\"finished_install\"):\n",
+        "    print(\"installing esmfold...\")\n",
         "    # install libs\n",
         "    os.system(\"pip install -q omegaconf pytorch_lightning biopython ml_collections einops py3Dmol\")\n",
         "    os.system(\"pip install -q git+https://github.com/NVIDIA/dllogger.git\")\n",
@@ -81,12 +82,12 @@
         "    os.system(\"touch finished_install\")\n",
         "\n",
         "  # wait for Params to finish downloading...\n",
-        "  if not os.path.isfile(model_name):\n",
-        "    print(\"ERROR: downloading esmfold params\")\n",
-        "  else:\n",
-        "    print(\"waiting for param download...\")\n",
-        "    while os.path.isfile(f\"{model_name}.aria2\"):\n",
-        "      time.sleep(5)"
+        "  while not os.path.isfile(model_name):\n",
+        "    time.sleep(5)\n",
+        "  if os.path.isfile(f\"{model_name}.aria2\"):\n",
+        "    print(\"downloading params...\")\n",
+        "  while os.path.isfile(f\"{model_name}.aria2\"):\n",
+        "    time.sleep(5)"
       ]
     },
     {