(self, text, normalized=False)
| 808 | return x @ self.text_projection |
| 809 | |
| 810 | def forward(self, text, normalized=False): |
| 811 | mask = dict() |
| 812 | |
| 813 | if self.l0_module is not None: |
| 814 | mask = self.l0_module.forward() |
| 815 | |
| 816 | self.mask = mask |
| 817 | |
| 818 | return self.encode_text(text, normalized=normalized, **mask) |
| 819 | |
| 820 | def prune(self): |
| 821 | device = self.token_embedding.weight.device |
nothing calls this directly
no test coverage detected