auracaster/piper
3
0
Fork 0
mirror of https://github.com/pstrueb/piper.git synced 2026-04-23 16:24:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Export without loading dataset

Browse Source
This commit is contained in:
Michael Hansen
2023-02-14 09:03:06 -06:00
parent 2da125a849
commit acc3068176
4 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/python/larynx_train/export_torchscript.py
View File

@@ -37,7 +37,7 @@ def main():
args.output = Path(args.output)
args.output.parent.mkdir(parents=True, exist_ok=True)
model = VitsModel.load_from_checkpoint(args.checkpoint)
model = VitsModel.load_from_checkpoint(args.checkpoint, dataset=None)
model_g = model.model_g
num_symbols = model_g.n_vocab