Force reinstall for hivemind.

2 years ago · 0570cc0e04
parent 0d9c7de0bd
commit 0570cc0e04
2 changed files with 4 additions and 0 deletions
--- a/examples/prompt-tuning-personachat.ipynb
+++ b/examples/prompt-tuning-personachat.ipynb
@ -42,6 +42,8 @@
    "if IN_COLAB:\n",
    "    subprocess.run(\"git clone https://github.com/bigscience-workshop/petals\", shell=True)\n",
    "    subprocess.run(\"pip install -r petals/requirements.txt\", shell=True)\n",
+    "    subprocess.run(\"pip uninstall -y hivemind\", shell=True)\n",
+    "    subprocess.run(\"pip install git+https://github.com/learning-at-home/hivemind@94c985d2dc7a79a091e46c755e9f2f4469b164c7\", shell=True)\n",
    "    subprocess.run(\"pip install datasets wandb\", shell=True)\n",
    "\n",
    "    try:\n",
--- a/examples/prompt-tuning-sst2.ipynb
+++ b/examples/prompt-tuning-sst2.ipynb
@ -42,6 +42,8 @@
    "if IN_COLAB:\n",
    "    subprocess.run(\"git clone https://github.com/bigscience-workshop/petals\", shell=True)\n",
    "    subprocess.run(\"pip install -r petals/requirements.txt\", shell=True)\n",
+    "    subprocess.run(\"pip uninstall -y hivemind\", shell=True)\n",
+    "    subprocess.run(\"pip install git+https://github.com/learning-at-home/hivemind@94c985d2dc7a79a091e46c755e9f2f4469b164c7\", shell=True)\n",
    "    subprocess.run(\"pip install datasets wandb\", shell=True)\n",
    "\n",
    "    try:\n",