From 789bb5d11be0c9f82203cbd9291aae42b36d5d3a Mon Sep 17 00:00:00 2001
From: mrq <mrq@ecker.tech>
Date: Sat, 13 Apr 2024 12:43:35 -0500
Subject: [PATCH] add an optional label override for model loading (used for
 easy testing between 12/16/20/24 layered model)

---
 vall_e/config.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/vall_e/config.py b/vall_e/config.py
index cdf3ac6..cef0bec 100755
--- a/vall_e/config.py
+++ b/vall_e/config.py
@@ -185,7 +185,10 @@ class Model:
 	def full_name(self):
 		name = [ self.name ]
 		
-		if self.size != "full" and isinstance(self.size, str):
+		if isinstance(self.size, dict):
+			if hasattr(self.size, "label") and self.size['label']:
+				name.append(f"{self.size['label']}")
+		elif isinstance(self.size, str) and self.size != "full":
 			name.append(self.size)
 
 		if self.arch_type != "transformer":
@@ -247,8 +250,6 @@ class Model:
 
 		if self.size == "double":
 			return 24
-		if self.size == "extended":
-			return 16
 		return 12
 
 	@property