feat: Add the ESM2 protein embedding model

README.md

@@ -154,6 +154,25 @@ embeddings = list(model.embed(images))
 # ]
 ```
 
+### 🧬 Protein embeddings
+
+```python
+from fastembed import ProteinEmbedding
+
+sequences = [
+    "MKTVRQERLKSIVRILERSKEPVSGAQLAEELSVSRQVIVQDIAYLRSLGYNIVATPRGYVLAGG",
+    "GKGDPKKPRGKMSSYAFFVQTSREEHKKKHPDASVNFSEFSKKCSERWKTMSAKEKGKFEDMAK",
+]
+
+model = ProteinEmbedding(model_name="facebook/esm2_t12_35M_UR50D")
+embeddings = list(model.embed(sequences))
+
+# [
+#   array([-0.0055, -0.0144,  0.0355, -0.0049, ...], dtype=float32),
+#   array([ 0.0114,  0.0020, -0.0247,  0.0060, ...], dtype=float32)
+# ]
+```
+
 ### Late interaction multimodal models (ColPali)
 
 ```python

fastembed/__init__.py

@@ -1,5 +1,6 @@
 import importlib.metadata
 
+from fastembed.bio import ProteinEmbedding
 from fastembed.image import ImageEmbedding
 from fastembed.late_interaction import LateInteractionTextEmbedding
 from fastembed.late_interaction_multimodal import LateInteractionMultimodalEmbedding
@@ -19,4 +20,5 @@
     "ImageEmbedding",
     "LateInteractionTextEmbedding",
     "LateInteractionMultimodalEmbedding",
-]
+    "ProteinEmbedding",
+]

fastembed/bio/__init__.py

@@ -0,0 +1,3 @@
+from fastembed.bio.protein_embedding import ProteinEmbedding
+
+__all__ = ["ProteinEmbedding"]