pin tf version #75

unitaryai · Dec 19, 2022 · f814c1a · f814c1a
1 parent 452932a
commit f814c1a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -7,15 +7,15 @@
 
 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",
     author="Unitary",
     author_email="[email protected]",
     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",
     ],