From 695df826c2c4a7ed2c825a714f199d207160ea7d Mon Sep 17 00:00:00 2001 From: Artem Chumachenko Date: Fri, 25 Nov 2022 22:48:16 +0400 Subject: [PATCH] Force reinstall for hivemind in example notebooks (#88) --- examples/prompt-tuning-personachat.ipynb | 2 ++ examples/prompt-tuning-sst2.ipynb | 2 ++ 2 files changed, 4 insertions(+) diff --git a/examples/prompt-tuning-personachat.ipynb b/examples/prompt-tuning-personachat.ipynb index 3c74449..e25d28e 100644 --- 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", diff --git a/examples/prompt-tuning-sst2.ipynb b/examples/prompt-tuning-sst2.ipynb index ed1de31..424d4eb 100644 --- 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",