(text, normalized=False)
| 961 | |
| 962 | def use_teacher_text(self): |
| 963 | def teacher_text_encoder_fn(text, normalized=False): |
| 964 | teacher = self.get_teacher() |
| 965 | with torch.no_grad(): |
| 966 | return teacher.encode_text(text, normalized=normalized) |
| 967 | self._text_encoder = FNBlock(teacher_text_encoder_fn) |
| 968 | |
| 969 | class EmptyTransformer(nn.Module): |
nothing calls this directly
no test coverage detected