mrq/tortoise-tts
4
6
Fork 15
Code Issues 9 Pull Requests 1 Packages Projects Releases Wiki Activity

backport

Browse Source
This commit is contained in:
mrq 2024-07-22 20:56:45 -05:00
parent f25e765682
commit a05faf0dfa
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tortoise_tts/export.py
View File

@ -50,6 +50,9 @@ def main():
if args.dtype != "auto": if args.dtype != "auto":
cfg.trainer.weight_dtype = args.dtype cfg.trainer.weight_dtype = args.dtype
# necessary to ensure we are actually exporting the weights right
cfg.inference.backend if inferencing else cfg.trainer.backend
engines = load_engines(training=False) engines = load_engines(training=False)
engines.export(userdata={"symmap": get_phone_symmap()}, callback=callback) engines.export(userdata={"symmap": get_phone_symmap()}, callback=callback)