lhallee commited on
Commit
ce0e094
·
verified ·
1 Parent(s): cc328db

Upload modeling_e1.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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