From f814c1a3803e8121fbd8aecaec456eea1fefcbf4 Mon Sep 17 00:00:00 2001
From: Laura Hanu <32672979+laurahanu@users.noreply.github.com>
Date: Mon, 19 Dec 2022 16:49:38 +0000
Subject: [PATCH] pin tf version #75

---
 requirements.txt | 2 +-
 setup.py         | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index e67590c..e05a574 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
 pytorch-lightning > 1.5.0
 torch >= 1.10.0
-transformers != 4.18.0
+transformers == 4.22.1 # pin to this version due to #75
 kaggle >= 1.5.8
 pandas >= 1.1.2
 scikit-learn >= 0.23.2
diff --git a/setup.py b/setup.py
index b30fbd0..fff7d04 100644
--- a/setup.py
+++ b/setup.py
@@ -7,7 +7,7 @@
 
 setup(
     name="detoxify",
-    version="0.5.0",
+    version="0.5.1",
     description="A python library for detecting toxic comments",
     long_description=long_description,
     long_description_content_type="text/markdown",
@@ -15,7 +15,7 @@
     author_email="laura@unitary.ai",
     url="https://github.com/unitaryai/detoxify",
     install_requires=[
-        "transformers != 4.18.0",  # v4.18.0 fails to properly load the finetuned weights
+        "transformers == 4.22.1", # pin to this version due to #75
         "torch >= 1.7.0",
         "sentencepiece >= 0.1.94",
     ],