auracaster/piper
3
0
Fork 0
mirror of https://github.com/pstrueb/piper.git synced 2026-04-20 07:14:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Don't load dataset during inference

Browse Source
This commit is contained in:
Michael Hansen
2023-02-23 09:22:55 -06:00
parent e4b6069e10
commit 131dbf4b5b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -29,7 +29,7 @@ def main():
args.output_dir = Path(args.output_dir)
args.output_dir.mkdir(parents=True, exist_ok=True)
model = VitsModel.load_from_checkpoint(args.checkpoint)
model = VitsModel.load_from_checkpoint(args.checkpoint, dataset=None)
# Inference only
model.eval()