auracaster/piper
3
0
Fork 0
mirror of https://github.com/pstrueb/piper.git synced 2026-04-18 06:15:30 +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_generator.py
View File

@@ -35,7 +35,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
# Inference only

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

@@ -39,7 +39,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

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

4
src/python/larynx_train/vits/lightning.py
View File

@@ -124,6 +124,10 @@ class VitsModel(pl.LightningModule):
num_test_examples: int,
max_phoneme_ids: Optional[int] = None,
):
if self.hparams.dataset is None:
_LOGGER.debug("No dataset to load")
return
full_dataset = LarynxDataset(
self.hparams.dataset, max_phoneme_ids=max_phoneme_ids
)