def forward(self, **kwargs):
    encoder_out = self.encoder(tbc=False, **kwargs)
    decoder_out = self.decoder(encoder_out=encoder_out, **kwargs)
    return decoder_out
def forward(self, **kwargs):
    encoder_out = self.encoder(tbc=False, **kwargs)
    decoder_out = self.decoder(encoder_out=encoder_out, **kwargs)
    return decoder_out