Upload modeling_e1.py with huggingface_hub
Browse files- modeling_e1.py +1 -0
modeling_e1.py
CHANGED
|
@@ -1515,6 +1515,7 @@ class EmbeddingMixin:
|
|
| 1515 |
save: bool = True,
|
| 1516 |
sql_db_path: str = 'embeddings.db',
|
| 1517 |
save_path: str = 'embeddings.pth',
|
|
|
|
| 1518 |
) -> Optional[dict[str, torch.Tensor]]:
|
| 1519 |
"""Embed a dataset of protein sequences.
|
| 1520 |
|
|
|
|
| 1515 |
save: bool = True,
|
| 1516 |
sql_db_path: str = 'embeddings.db',
|
| 1517 |
save_path: str = 'embeddings.pth',
|
| 1518 |
+
**kwargs,
|
| 1519 |
) -> Optional[dict[str, torch.Tensor]]:
|
| 1520 |
"""Embed a dataset of protein sequences.
|
| 1521 |
|