mrq/vall-e
5
9
Fork 0
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

might help

Browse Source
This commit is contained in:
mrq 2024-07-22 20:57:01 -05:00
parent 188d116222
commit 611a1c4bdc
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
vall_e/export.py
View File

@ -127,6 +127,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 = 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)