Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update requirements.txt
Browse files- requirements.txt +7 -4
requirements.txt
CHANGED
@@ -1,12 +1,13 @@
|
|
1 |
# deep learning
|
2 |
# Install with conda
|
3 |
-
|
4 |
-
|
5 |
-
lightning>=2.
|
6 |
torchmetrics
|
7 |
transformers
|
8 |
scikit-learn
|
9 |
dgl
|
|
|
10 |
dgllife
|
11 |
|
12 |
# configuration and utilities
|
@@ -14,17 +15,19 @@ hydra-core
|
|
14 |
pandas
|
15 |
swifter
|
16 |
tensorboard
|
|
|
17 |
|
18 |
# model-specific requirements
|
19 |
subword-nmt # MolTrans
|
20 |
tape_proteins # TransformerCPI2
|
|
|
21 |
fair-esm # CoaDTI-pro
|
22 |
# apex # recommended by TAPE-BERT
|
23 |
|
24 |
# feature extraction
|
25 |
rdkit
|
26 |
mordred
|
27 |
-
networkx>=3.
|
28 |
mhfp
|
29 |
panel
|
30 |
plotly
|
|
|
1 |
# deep learning
|
2 |
# Install with conda
|
3 |
+
torch >= 2.1
|
4 |
+
torch_geometric
|
5 |
+
lightning >= 2.1
|
6 |
torchmetrics
|
7 |
transformers
|
8 |
scikit-learn
|
9 |
dgl
|
10 |
+
dglgo
|
11 |
dgllife
|
12 |
|
13 |
# configuration and utilities
|
|
|
15 |
pandas
|
16 |
swifter
|
17 |
tensorboard
|
18 |
+
biopython
|
19 |
|
20 |
# model-specific requirements
|
21 |
subword-nmt # MolTrans
|
22 |
tape_proteins # TransformerCPI2
|
23 |
+
gensim
|
24 |
fair-esm # CoaDTI-pro
|
25 |
# apex # recommended by TAPE-BERT
|
26 |
|
27 |
# feature extraction
|
28 |
rdkit
|
29 |
mordred
|
30 |
+
networkx >= 3.2
|
31 |
mhfp
|
32 |
panel
|
33 |
plotly
|