From a05faf0dfa9bda546b78fccdad41420a491de856 Mon Sep 17 00:00:00 2001
From: mrq <mrq@ecker.tech>
Date: Mon, 22 Jul 2024 20:56:45 -0500
Subject: [PATCH] backport

---
 tortoise_tts/export.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tortoise_tts/export.py b/tortoise_tts/export.py
index 8af9c7a..f6d3ce2 100755
--- a/tortoise_tts/export.py
+++ b/tortoise_tts/export.py
@@ -49,7 +49,10 @@ 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 if inferencing else cfg.trainer.backend
+	
 	engines = load_engines(training=False)
 	engines.export(userdata={"symmap": get_phone_symmap()}, callback=callback)