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

This commit is contained in:
mrq 2024-04-13 12:43:35 -05:00
parent f0c4baeb25
commit 789bb5d11b

View File

@ -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