def METHOD_NAME(self, fastpitch_model):
    model = fastpitch_model.cuda()
    with tempfile.TemporaryDirectory() as tmpdir:
        filename = os.path.join(tmpdir, 'fp.onnx')
        model.export(output=filename, verbose=True, onnx_opset_version=14, check_trace=True)