(self, input)
| 106 | return torch.gather(softmax, 1, best_id[..., None, None].expand(best_id.shape[0], len(self.negatives), 2))[:, 0, :] |
| 107 | |
| 108 | def encode_image(self, input): |
| 109 | processed_input = self.process(input).half() |
| 110 | return self.model.encode_image(processed_input) |
| 111 | |
| 112 | |
| 113 |
no outgoing calls
no test coverage detected