def __init__(self, text_encoder: BaseTextEncoder, *args, **kwargs):
    super().__init__(*args, **kwargs)
    self.text_encoder = text_encoder
def __init__(self, text_encoder: BaseTextEncoder, *args, **kwargs):
    super().__init__(*args, **kwargs)
    self.text_encoder = text_encoder