(self, text)
| 106 | return embed.float() |
| 107 | |
| 108 | def tokenizer(self, text): |
| 109 | result = self.tokenize( |
| 110 | text, |
| 111 | padding="max_length", |
| 112 | truncation=True, |
| 113 | max_length=512, |
| 114 | return_tensors="pt", |
| 115 | ) |
| 116 | return {k: v.squeeze(0) for k, v in result.items()} |