| """Text encoder.""" | |
| from sentence_transformers import SentenceTransformer | |
| from torch import Tensor | |
| class TextEncoder: | |
| """Text encoder.""" | |
| def __init__(self) -> None: | |
| """Initialize the text encoder.""" | |
| self.model = SentenceTransformer("sentence-transformers/all-MiniLM-L12-v1") | |
| def encode(self, text: str) -> Tensor: | |
| """Encode a string into an embedding.""" | |
| return self.model.encode(text) | |