inidicate we only support pytorch 1.0.0+ now

2019-08-05 14:38:59 +02:00
parent 328afb7097
commit 58830807d1
3 changed files with 3 additions and 3 deletions
--- a/setup.py
+++ b/setup.py
@@ -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',