Updating to Transformers v2.6.0 (#1059)

* Transformers v2.6.0 * requirements update
nyu-mll · Apr 22, 2020 · dae9dc6 · dae9dc6
1 parent 67238af
commit dae9dc6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/environment.yml b/environment.yml
@@ -39,8 +39,9 @@ dependencies:
         - sacremoses
 
         # Warning: jiant currently depends on *both* pytorch_pretrained_bert > 0.6 _and_
-        # transformers > 2.3.0. These are the same package, though the name changed between
+        # transformers > 2.6.0. These are the same package, though the name changed between
         # these two versions. AllenNLP requires 0.6 to support the BertAdam optimizer, and jiant
         # directly requires 1.0 to support XLNet and WWM-BERT.
         # This AllenNLP issue is relevant: https://github.com/allenai/allennlp/issues/3067
-        - transformers==2.3.0
+        - transformers==2.6.0
+        - tokenizers==0.5.2
diff --git a/setup.py b/setup.py
@@ -65,7 +65,8 @@
         "pyhocon==0.3.35",
         "python-Levenshtein==0.12.0",
         "sacremoses",
-        "transformers==2.3.0",
+        "transformers==2.6.0",
+        "tokenizers==0.5.2",
         "ftfy",
         "spacy",
     ],