Upload modeling_fastesm.py with huggingface_hub
Browse files- modeling_fastesm.py +1 -0
modeling_fastesm.py
CHANGED
|
@@ -679,6 +679,7 @@ class EmbeddingMixin:
|
|
| 679 |
save: bool = True,
|
| 680 |
sql_db_path: str = 'embeddings.db',
|
| 681 |
save_path: str = 'embeddings.pth',
|
|
|
|
| 682 |
) -> Optional[dict[str, torch.Tensor]]:
|
| 683 |
"""Embed a dataset of protein sequences.
|
| 684 |
|
|
|
|
| 679 |
save: bool = True,
|
| 680 |
sql_db_path: str = 'embeddings.db',
|
| 681 |
save_path: str = 'embeddings.pth',
|
| 682 |
+
**kwargs,
|
| 683 |
) -> Optional[dict[str, torch.Tensor]]:
|
| 684 |
"""Embed a dataset of protein sequences.
|
| 685 |
|