(self, query_list: Union[List[str], str], batch_size=64, is_query=True)
| 74 | |
| 75 | @torch.inference_mode() |
| 76 | def multi_gpu_encode(self, query_list: Union[List[str], str], batch_size=64, is_query=True) -> np.ndarray: |
| 77 | if self.gpu_num > 1: |
| 78 | self.model = torch.nn.DataParallel(self.model) |
| 79 | query_emb = self.encode(query_list, batch_size, is_query) |
| 80 | return query_emb |
| 81 | |
| 82 | |
| 83 | class STEncoder: |
no test coverage detected