"""Export Piper with its legacy dynamic-axis ONNX exporter on newer PyTorch.""" import functools import torch from piper.train.export_onnx import main # Piper's exporter specifies dynamic_axes/opset 15 and predates the dynamo default. torch.onnx.export = functools.partial(torch.onnx.export, dynamo=False) if __name__ == "__main__": main()