From 31cfef59c473019d2824ffaa2a37fe4b9e2d470d Mon Sep 17 00:00:00 2001
From: mrq <mrq@ecker.tech>
Date: Fri, 27 Dec 2024 18:16:57 -0600
Subject: [PATCH] when you do more training thinking the original model that
 can do NS/SR got deleted but it was actually a string not having its quotes
 in the right place.......

---
 vall_e/inference.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vall_e/inference.py b/vall_e/inference.py
index 9a9c8d1..d774caf 100755
--- a/vall_e/inference.py
+++ b/vall_e/inference.py
@@ -436,7 +436,7 @@ class TTS():
 		prom = self.encode_audio( references, trim_length=input_prompt_length ) if references else None
 		lang = self.encode_lang( language )
 		
-		if task in ["ns, sr"]:
+		if task in ["ns", "sr"]:
 			prom = [
 				task,
 				prom