diff --git a/README.md b/README.md index 6e95999ca..1c87b8a72 100644 --- a/README.md +++ b/README.md @@ -37,7 +37,7 @@ The following publications are integrated in this framework: ## Installation -We recommend **Python 3.6** or higher, **[PyTorch 1.6.0](https://pytorch.org/get-started/locally/)** or higher and **[transformers v4.6.0](https://github.com/huggingface/transformers)** or higher. The code does **not** work with Python 2.7. +We recommend **Python 3.6** or higher, **[PyTorch 1.6.0](https://pytorch.org/get-started/locally/)** or higher and **[transformers v4.34.1](https://github.com/huggingface/transformers)** or lower. The code does **not** work with Python 2.7. **Install with pip** diff --git a/requirements.txt b/requirements.txt index 3d52c08e8..f7505eb28 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -transformers>=4.6.0,<5.0.0 +transformers<4.35.0 tokenizers>=0.10.3 tqdm torch>=1.6.0 diff --git a/setup.py b/setup.py index 995727a71..380cf9a87 100644 --- a/setup.py +++ b/setup.py @@ -19,7 +19,7 @@ packages=find_packages(), python_requires=">=3.6.0", install_requires=[ - 'transformers>=4.6.0,<5.0.0', + 'transformers<4.35.0', 'tqdm', 'torch>=1.6.0', 'torchvision',