From f3fafd14a4c00448c67d028c6065e2bef2520fa5 Mon Sep 17 00:00:00 2001
From: Alexander Borzunov <borzunov.alexander@gmail.com>
Date: Sun, 23 Jul 2023 18:45:19 +0400
Subject: [PATCH] Bump version to 2.0.1 (#411)

---
 setup.cfg              | 4 ++--
 src/petals/__init__.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/setup.cfg b/setup.cfg
index 417a126a5..7c04686fb 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -38,8 +38,8 @@ install_requires =
     tokenizers>=0.13.3
     transformers>=4.31.0,<5.0.0
     speedtest-cli==2.1.3
-    pydantic>=1.10,<2.0  # 2.0 is incompatible with hivemind==1.1.8
-    hivemind @ git+https://github.com/learning-at-home/hivemind
+    pydantic>=1.10,<2.0  # 2.0 is incompatible with hivemind yet
+    hivemind==1.1.9
     tensor_parallel==1.0.23
     humanfriendly
     async-timeout>=4.0.2
diff --git a/src/petals/__init__.py b/src/petals/__init__.py
index c696bc682..d2f91bcb1 100644
--- a/src/petals/__init__.py
+++ b/src/petals/__init__.py
@@ -11,7 +11,7 @@
 from petals.utils import *
 from petals.utils.logging import initialize_logs as _initialize_logs
 
-__version__ = "2.0.0.post3"
+__version__ = "2.0.1"
 
 
 if not os.getenv("PETALS_IGNORE_DEPENDENCY_VERSION"):