Encode with query prompt for single items, document prompt for batches.
(self, texts: Sequence[str], /)
| 81 | self._model.max_seq_length = max_seq_length |
| 82 | |
| 83 | def encode(self, texts: Sequence[str], /) -> EmbeddingMatrix: |
| 84 | """Encode with query prompt for single items, document prompt for batches.""" |
| 85 | text_list = list(texts) |
| 86 | if len(text_list) == 1: |
| 87 | return self._model.encode(text_list, prompt_name="query", batch_size=1) # type: ignore[return-value] |
| 88 | return self._model.encode(text_list, batch_size=1) # type: ignore[return-value] |
| 89 | |
| 90 | |
| 91 | def _bench_semble( |
no outgoing calls