jannisborn commited on
Commit
794cabf
1 Parent(s): c911d89
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -8,6 +8,7 @@ torch-sparse
8
  torch-geometric
9
  torchvision==0.13.1
10
  torchaudio==0.12.1
 
11
  gt4sd>=1.1.6
12
  diffusers==0.6.0
13
  molgx>=0.22.0a1
@@ -23,7 +24,7 @@ numpy==1.23.5
23
  pandas>=1.0.0,<2.0.0
24
  httpcore==0.15
25
  pydantic<2.0.0
26
- accelerate==0.16.0
27
  terminator @ git+https://github.com/IBM/regression-transformer@gt4sd
28
  guacamol_baselines @ git+https://github.com/GT4SD/guacamol_baselines.git@v0.0.2
29
  moses @ git+https://github.com/GT4SD/moses.git@v0.1.0
 
8
  torch-geometric
9
  torchvision==0.13.1
10
  torchaudio==0.12.1
11
+ torchmetrics==0.11.4
12
  gt4sd>=1.1.6
13
  diffusers==0.6.0
14
  molgx>=0.22.0a1
 
24
  pandas>=1.0.0,<2.0.0
25
  httpcore==0.15
26
  pydantic<2.0.0
27
+ pytorch-lightning==1.7.7
28
  terminator @ git+https://github.com/IBM/regression-transformer@gt4sd
29
  guacamol_baselines @ git+https://github.com/GT4SD/guacamol_baselines.git@v0.0.2
30
  moses @ git+https://github.com/GT4SD/moses.git@v0.1.0