change the download source of uf symmetry parameters to google drive (#…

…51)
dptech-corp · Sep 16, 2022 · 1624ef2 · 1624ef2
1 parent 9e47a64
commit 1624ef2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/notebooks/unifold.ipynb b/notebooks/unifold.ipynb
@@ -83,7 +83,7 @@
         "GIT_REPO='https://github.com/dptech-corp/Uni-Fold'\n",
         "UNICORE_URL='https://github.com/dptech-corp/Uni-Core/releases/download/0.0.1/unicore-0.0.1+cu113torch1.12.1-cp37-cp37m-linux_x86_64.whl'\n",
         "PARAM_URL='https://drive.google.com/uc?id=1A9iXMYCwP0f_U0FgISJ_6BX7FXZtglvV'\n",
-        "UF_SYMM_PARAM_URL='https://uni-fold.dp.tech/uf_symmetry_params_2022-09-06.tar.gz' # TODO: use Google drive.\n",
+        "UF_SYMM_PARAM_URL='https://drive.google.com/uc?id=1UNEGzmueQTxY05QIRweKHxOjr1ht-G_Q'\n",
         "\n",
         "if [ ! -f UNIFOLD_READY ]; then\n",
         "    wget  ${UNICORE_URL} \n",
@@ -92,7 +92,7 @@
         "    pip3 -q install ./Uni-Fold\n",
         "    gdown ${PARAM_URL}\n",
         "    tar -xzf \"unifold_params_2022-08-01.tar.gz\"\n",
-        "    wget  ${UF_SYMM_PARAM_URL}\n",
+        "    gdown  ${UF_SYMM_PARAM_URL}\n",
         "    tar -xzf \"uf_symmetry_params_2022-09-06.tar.gz\"\n",
         "\n",
         "    touch UNIFOLD_READY\n",