dev(hansbug): try add support of torch 1.10

opendilab · Jan 4, 2022 · fe5f681 · fe5f681
1 parent ff136da
commit fe5f681
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -25,6 +25,7 @@ jobs:
           - '1.7.0'
           - '1.8.0'
           - '1.9.0'
+          - '1.10.0'
         exclude:
           - os: 'ubuntu-18.04'
             python-version: '3.9'

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
 treevalue>=1.2.0
-torch>=1.3.1,<=1.9.0
+torch>=1.1.0,<=1.10.0
 hbutils>=0.0.1
 numpy