From 2c21a0edac9f57141967814f785c841eb95bcf5b Mon Sep 17 00:00:00 2001 From: mush42 Date: Sun, 24 Sep 2023 20:28:59 +0200 Subject: [PATCH] Fixed an error encountered when loading the vocoder during export. --- matcha/onnx/export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/matcha/onnx/export.py b/matcha/onnx/export.py index 6fc0765..9b79508 100644 --- a/matcha/onnx/export.py +++ b/matcha/onnx/export.py @@ -127,7 +127,7 @@ def main(): assert ( args.vocoder_name and args.vocoder_checkpoint_path ), "Both vocoder_name and vocoder-checkpoint are required when embedding the vocoder in the ONNX graph." - vocoder = load_vocoder(args.vocoder_name, args.vocoder_checkpoint_path, "cpu") + vocoder, _ = load_vocoder(args.vocoder_name, args.vocoder_checkpoint_path, "cpu") else: vocoder = None