Browse Source

Force reinstall for hivemind in example notebooks (#88)

Artem Chumachenko 2 years ago
parent
commit
695df826c2
2 changed files with 4 additions and 0 deletions
  1. 2 0
      examples/prompt-tuning-personachat.ipynb
  2. 2 0
      examples/prompt-tuning-sst2.ipynb

+ 2 - 0
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",

+ 2 - 0
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",