Force reinstall for hivemind in example notebooks (#88)

2 years ago · 695df826c2
parent dc6ecccac5
commit 695df826c2
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",