Download boltz weights from R2

sokrypton · Nov 18, 2024 · 15050d8 · 15050d8
1 parent 7932813
commit 15050d8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/Boltz1.ipynb b/Boltz1.ipynb
@@ -210,7 +210,11 @@
         "  os.system(\"touch COLABFOLD_READY\")\n",
         "\n",
         "if not os.path.isfile(\"BOLZ_READY\"):\n",
+        "  os.system(\"apt-get install -y aria2\")\n",
         "  os.system(\"pip install -q --no-warn-conflicts boltz\")\n",
+        "  os.system(\"mkdir weights\")\n",
+        "  os.system(\"aria2c -d weights -x8 -s8 https://colabfold.steineggerlab.workers.dev/boltz1.ckpt\")\n",
+        "  os.system(\"aria2c -d weights -x8 -s8 https://colabfold.steineggerlab.workers.dev/ccd.pkl\")\n",
         "  os.system(\"touch BOLZ_READY\")"
       ]
     },
@@ -231,7 +235,7 @@
       "cell_type": "code",
       "source": [
         "#@title Predict structure using boltz\n",
-        "!boltz predict --out_dir \"{jobname}\" \"{jobname}/{jobname}.fasta\""
+        "!boltz predict --out_dir \"{jobname}\" \"{jobname}/{jobname}.fasta\" --cache weights"
       ],
       "metadata": {
         "cellView": "form",