SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

inidicate we only support pytorch 1.0.0+ now

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -25,7 +25,7 @@ These implementations have been tested on several datasets (see the example scri
## Installation ## 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 ### With pip

2
requirements.txt
View File

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

2
setup.py
View File

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