def __init__(self, enc_attn_dim, vocab_size, use_checkpoint, dropout_text):
    super().__init__()
    self.text_encoder = TransformerEncoder(enc_attn_dim, vocab_size, use_checkpoint, clip=True, dropout=dropout_text)
def __init__(self, enc_attn_dim, vocab_size, use_checkpoint, dropout_text):
    super().__init__()
    self.text_encoder = TransformerEncoder(enc_attn_dim, vocab_size, use_checkpoint, clip=True, dropout=dropout_text)