(self, text, return_mask = False)
| 193 | } |
| 194 | |
| 195 | def encode_list(self, text, return_mask = False): |
| 196 | t5_embeds = self.t5_model.encode_list(text, return_mask = return_mask) |
| 197 | clip_embeds = self.clip_model.encode_list(text, return_mask = return_mask) |
| 198 | # change embedding strategy here |
| 199 | return { |
| 200 | 'context': t5_embeds, |
| 201 | 'y': clip_embeds, |
| 202 | } |
| 203 | |
| 204 | def encode_list_of_list(self, text, return_mask = False): |
| 205 | t5_embeds = self.t5_model.encode_list_of_list(text, return_mask = return_mask) |
nothing calls this directly
no test coverage detected