lhallee commited on
Commit
8f5693f
·
verified ·
1 Parent(s): 1b59c7f

Upload modeling_esm_plusplus.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_esm_plusplus.py +1 -0
modeling_esm_plusplus.py CHANGED
@@ -685,6 +685,7 @@ class EmbeddingMixin:
685
  save: bool = True,
686
  sql_db_path: str = 'embeddings.db',
687
  save_path: str = 'embeddings.pth',
 
688
  ) -> Optional[dict[str, torch.Tensor]]:
689
  """Embed a dataset of protein sequences.
690
 
 
685
  save: bool = True,
686
  sql_db_path: str = 'embeddings.db',
687
  save_path: str = 'embeddings.pth',
688
+ **kwargs,
689
  ) -> Optional[dict[str, torch.Tensor]]:
690
  """Embed a dataset of protein sequences.
691