update torch requirement

allenai · Oct 27, 2020 · 98edd25 · 98edd25
2 parents 8da3508 + 5d6670c
commit 98edd25
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -52,6 +52,7 @@ data loaders.  Those are coming soon.
   You can also now set default values for foo like `foo: Lazy[Foo] = Lazy(Foo)`.
   Or, if you want you want a default value but also want to allow for `None` values, you can
   write it like this: `foo: Optional[Lazy[Foo]] = Lazy(Foo)`.
+- Added support for PyTorch version 1.7.
 
 ### Fixed
 

diff --git a/setup.py b/setup.py
@@ -50,7 +50,7 @@
         ]
     ),
     install_requires=[
-        "torch>=1.6.0,<1.7.0",
+        "torch>=1.6.0,<1.8.0",
         "torchvision>=0.7.0",
         "jsonnet>=0.10.0 ; sys.platform != 'win32'",
         "overrides==3.1.0",