Skip to content

Commit 401fac7

Browse files
(fix) update tags with device argument
1 parent 9fa20a0 commit 401fac7

3 files changed

Lines changed: 8 additions & 9 deletions

File tree

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -88,9 +88,9 @@ It will be really useful to retrieve it from deepchain hub.
8888
- librairies
8989
- embeddings
9090
- datasets
91-
- cuda
91+
- device
9292

93-
If you want your app to benefit from deepchain' GPU, set cuda to "True" in tags. It will run on CPU by default.
93+
If you want your app to benefit from deepchain' GPU, set device to "gpu" in tags. It will run on "cpu" by default.
9494

9595
### Special method in app
9696

src/DESC.md

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
# Description
22
This app is a base application that compute the loglikelihood (the probability that a protein is natural or not).
33

4-
We use a bio-transformers with a ESM 'prot_bert' model as backend to make the computation of logits
4+
We use a bio-transformers with a 'protbert' model as backend to make the computation of logits
55

66
# Tags
77
Fill the tags.json file in this folder:
@@ -10,8 +10,8 @@ Fill the tags.json file in this folder:
1010
- libraries: str -> libraries used for the app
1111
- embeddings: str -> embedding model if used.
1212
- datasets: str -> dataset name if used.
13-
- cuda: []: bool -> By default, the app are launch on CPU on deepchain, put cuda to "True" to benefit GPU
14-
and accelerate the optimization process during score computation.
13+
- device: ["cpu"]: str -> By default, the app are launched on cpu on deepchain, put device to "gpu" to benefit GPU
14+
and accelerate the optimization process during score computation.
1515

1616

1717
## libraries
@@ -25,5 +25,4 @@ Fill the tags.json file in this folder:
2525
## embeddings
2626
- ESM
2727

28-
## datasets
29-
28+
## datasets

src/tags.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
"libraries": [],
44
"embeddings": [],
55
"datasets": [],
6-
"cuda": [
7-
"False"
6+
"device": [
7+
"cpu"
88
]
99
}

0 commit comments

Comments
 (0)