Commit e2879a17 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Pinning back to torch 1.6.*

parent 52c55668
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -4,4 +4,4 @@ sphinx>=3.2,<4
sphinx_rtd_theme>=0.5,<1
tensorflow>=2.4
transformers
torch==1.8.0
torch==1.6.0
+1 −1
Original line number Diff line number Diff line
@@ -25,7 +25,7 @@ dependencies:
    - tensorflow~=2.5
    - tensorflow_probability==0.12.*
    - tensorflow_addons==0.12.*
    - torch-geometric==1.7.0
    - torch-geometric==1.6.0
    - transformers==4.6.*
    - xgboost==1.*
    - git+https://github.com/samoturk/mol2vec
+7 −5
Original line number Diff line number Diff line
dependencies:
  - pip:
    - -f https://pytorch-geometric.com/whl/torch-1.8.0+cpu.html
    - -f https://pytorch-geometric.com/whl/torch-1.6.0+cpu.html
    - dgl==0.6.*
    - torch==1.8.1
    - torchvision==0.9.1
    - torch-scatter
    - torch-sparse
    - torch==1.6.0
    - torchvision==0.7.0
    - torch-cluster==1.5.9
    - torch-scatter==2.0.6
    - torch-sparse==0.6.9
    - torch-spline-conv==1.2.1
+7 −5
Original line number Diff line number Diff line
dependencies:
  - pip:
    - -f https://download.pytorch.org/whl/torch_stable.html
    - -f https://pytorch-geometric.com/whl/torch-1.8.0+cpu.html
    - -f https://pytorch-geometric.com/whl/torch-1.6.0+cpu.html
    - dgl==0.6.*
    - torch==1.8.1+cpu
    - torchvision==0.9.1+cpu
    - torch-scatter
    - torch-sparse
    - torch==1.6.0+cpu
    - torchvision==0.7.0+cpu
    - torch-cluster==1.5.9
    - torch-scatter==2.0.6
    - torch-sparse==0.6.9
    - torch-spline-conv==1.2.1
+8 −6
Original line number Diff line number Diff line
dependencies:
  - pip:
    - -f https://download.pytorch.org/whl/torch_stable.html
    - -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
    - dgl-cu110==0.6.*
    - torch==1.8.0+cu111
    - torchvision==0.9.1+cu111
    - torch-scatter
    - torch-sparse
    - -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html
    - dgl-cu101==0.6.*
    - torch==1.6.0+cu101
    - torchvision==0.7.0+cu101
    - torch-cluster==1.5.9
    - torch-scatter==2.0.6
    - torch-sparse==0.6.9
    - torch-spline-conv==1.2.1