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 ( assert (
args.vocoder_name and args.vocoder_checkpoint_path args.vocoder_name and args.vocoder_checkpoint_path
), "Both vocoder_name and vocoder-checkpoint are required when embedding the vocoder in the ONNX graph." ), "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: else:
vocoder = None vocoder = None