diff --git a/README.md b/README.md index fda2076..5deac8a 100644 --- a/README.md +++ b/README.md @@ -78,7 +78,7 @@ predefined_args = { * gluonnlp >= 0.6.0 * sentencepiece >= 0.1.6 * onnxruntime >= 0.3.0 -* transformers >= 2.1.1 +* transformers >= 2.1.1, <= 3.0.2 #### How to install diff --git a/requirements.txt b/requirements.txt index 2422f23..f0a46e9 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,4 +3,4 @@ mxnet >= 1.4.0 gluonnlp >= 0.6.0 sentencepiece >= 0.1.6 onnxruntime >= 0.3.0 -transformers >= 2.1.1 +transformers >= 2.1.1, <= 3.0.2