jannisborn commited on
Commit
583fb30
1 Parent(s): e3a6457
Files changed (1) hide show
  1. requirements.txt +5 -2
requirements.txt CHANGED
@@ -6,7 +6,7 @@ torch-scatter
6
  torch-spline-conv
7
  torch-sparse
8
  torch-geometric
9
- torchvision==0.13.1
10
  torchaudio==0.12.1
11
  diffusers==0.6.0
12
  gt4sd==1.3.1
@@ -34,4 +34,7 @@ paccmann_predictor @ git+https://github.com/PaccMann/paccmann_predictor@sarscov2
34
  reinvent_models @ git+https://github.com/GT4SD/reinvent_models@v0.0.1
35
  h11==0.13.0
36
  pytorch-lightning==1.7.7
37
- torchmetrics==0.11.3
 
 
 
 
6
  torch-spline-conv
7
  torch-sparse
8
  torch-geometric
9
+ torchvision==0.14.1
10
  torchaudio==0.12.1
11
  diffusers==0.6.0
12
  gt4sd==1.3.1
 
34
  reinvent_models @ git+https://github.com/GT4SD/reinvent_models@v0.0.1
35
  h11==0.13.0
36
  pytorch-lightning==1.7.7
37
+ torchmetrics==0.11.3
38
+ lightning-utilities==0.9.0
39
+ scikit-learn==1.2.1
40
+ scikit-optimize==0.9.0