Upload modeling_esm_plusplus.py with huggingface_hub

modeling_esm_plusplus.py

@@ -597,7 +597,7 @@ class ProteinDataset(TorchDataset):
 def build_collator(tokenizer) -> Callable[[list[str]], tuple[torch.Tensor, torch.Tensor]]:
     def _collate_fn(sequences: list[str]) -> tuple[torch.Tensor, torch.Tensor]:
         """Collate function for batching sequences."""
-        return tokenizer(sequences, return_tensors="pt", padding='longest', pad_to_multiple_of=8)
+        return tokenizer(sequences, return_tensors="pt", padding='longest')
     return _collate_fn
 
 
@@ -755,7 +755,6 @@ class EmbeddingMixin:
 
         return embeddings_dict
 
-
 class PreTrainedESMplusplusModel(PreTrainedModel):
     """
     init weights for ESM++ models