| 953 | self._image_encoder = FNBlock(teacher_image_encoder_fn) |
| 954 | |
| 955 | class EmptyVisual(nn.Module): |
| 956 | def __init__(self): |
| 957 | super().__init__() |
| 958 | self.layers = 0 |
| 959 | self._image_encoder.visual = EmptyVisual() |
| 960 | self._without_ddp[0] = self._image_encoder |
| 961 |