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

add an optional label override for model loading (used for easy testing between 12/16/20/24 layered model)

Browse Source
This commit is contained in:
mrq 2024-04-13 12:43:35 -05:00
parent f0c4baeb25
commit 789bb5d11b
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
vall_e/config.py
View File

@ -185,7 +185,10 @@ class Model:
def full_name(self): def full_name(self):
name = [ self.name ] 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) name.append(self.size)
if self.arch_type != "transformer": if self.arch_type != "transformer":
@ -247,8 +250,6 @@ class Model:
if self.size == "double": if self.size == "double":
return 24 return 24
if self.size == "extended":
return 16
return 12 return 12
@property @property