mirror of
https://github.com/shivammehta25/Matcha-TTS.git
synced 2026-02-04 09:49:21 +08:00
Fixed an error encountered when loading the vocoder during export.
This commit is contained in:
@@ -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
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user