pin tf version #75

Author: laurahanu

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

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",
     ],