config.decoder.torchscript = True
config.encoder.torchscript = True