inidicate we only support pytorch 1.0.0+ now

This commit is contained in:
thomwolf
2019-08-05 14:38:59 +02:00
parent 328afb7097
commit 58830807d1
3 changed files with 3 additions and 3 deletions

View File

@@ -25,7 +25,7 @@ These implementations have been tested on several datasets (see the example scri
## Installation
This repo is tested on Python 2.7 and 3.5+ (examples are tested only on python 3.5+) and PyTorch 0.4.1 to 1.1.0
This repo is tested on Python 2.7 and 3.5+ (examples are tested only on python 3.5+) and PyTorch 1.0.0+
### With pip

View File

@@ -1,5 +1,5 @@
# PyTorch
torch>=0.4.1
torch>=1.0.0
# progress bars in model download and training scripts
tqdm
# Accessing files from S3 directly.

View File

@@ -49,7 +49,7 @@ setup(
url="https://github.com/huggingface/pytorch-transformers",
packages=find_packages(exclude=["*.tests", "*.tests.*",
"tests.*", "tests"]),
install_requires=['torch>=0.4.1',
install_requires=['torch>=1.0.0',
'numpy',
'boto3',
'requests',