might help

2024-07-22 20:57:01 -05:00 · 2024-07-22 20:57:01 -05:00 · 611a1c4bdc
commit 611a1c4bdc
parent 188d116222
1 changed files with 3 additions and 0 deletions
--- a/vall_e/export.py
+++ b/vall_e/export.py
@ -127,6 +127,9 @@ def main():
 	if args.dtype != "auto":
 		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.export(userdata={"symmap": get_phone_symmap()}, callback=callback)