Fixed an error encountered when loading the vocoder during export.

This commit is contained in:
mush42
2023-09-24 20:28:59 +02:00
parent 25767f76a8
commit 2c21a0edac

View File

@@ -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