Commit 58830807 authored by thomwolf's avatar thomwolf
Browse files

inidicate we only support pytorch 1.0.0+ now

parent 328afb70
......@@ -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
......
# PyTorch
torch>=0.4.1
torch>=1.0.0
# progress bars in model download and training scripts
tqdm
# Accessing files from S3 directly.
......
......@@ -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',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment