(self, prompts: Iterable[str])
| 251 | return self.model.embed_images(xs) |
| 252 | |
| 253 | def embed_text(self, prompts: Iterable[str]) -> torch.Tensor: |
| 254 | with torch.no_grad(): |
| 255 | return self.model.embed_text(prompts) |
| 256 | |
| 257 | def embed_images_grid(self, xs: Iterable[Optional[ImageType]]) -> torch.Tensor: |
| 258 | with torch.no_grad(): |
nothing calls this directly
no test coverage detected