Update README.md
Browse files
README.md
CHANGED
|
@@ -89,7 +89,8 @@ print(mut_dict)
|
|
| 89 |
|
| 90 |
## Get protein embeddings
|
| 91 |
If you want to generate protein embeddings, you could refer to the following code. The embeddings are the average of
|
| 92 |
-
the hidden states of the last layer.
|
|
|
|
| 93 |
```python
|
| 94 |
from model.saprot.base import SaprotBaseModel
|
| 95 |
from transformers import EsmTokenizer
|
|
|
|
| 89 |
|
| 90 |
## Get protein embeddings
|
| 91 |
If you want to generate protein embeddings, you could refer to the following code. The embeddings are the average of
|
| 92 |
+
the hidden states of the last layer.
|
| 93 |
+
<!-- <font color=red>Note frozen SaProt supports SA sequence embeddings but not AA sequence embeddings.</font> -->
|
| 94 |
```python
|
| 95 |
from model.saprot.base import SaprotBaseModel
|
| 96 |
from transformers import EsmTokenizer
|