def forward(self, y):
    text_features = self.text_encoder(y)
    return F.normalize(text_features, dim=(- 1))
def forward(self, y):
    text_features = self.text_encoder(y)
    return F.normalize(text_features, dim=(- 1))