-
🛈
-
@@ -322,6 +323,22 @@ class ExtraNetworksPage:
return '#{:02x}{:02x}{:02x}'.format(r, g, b) # pylint: disable=consider-using-f-string
try:
+ base_model = ''
+ base_model_box = ''
+ info = self.find_info(item.get('filename'))
+ item_hash = item.get('hash', None)
+ if info and item_hash and ('modelVersions' in info):
+ for version in info.get('modelVersions', []):
+ for file_info in version.get('files', []):
+ if any(h.lower().startswith(item_hash.lower()) for h in file_info.get('hashes', {}).values()):
+ base_model = version.get('baseModel', '')
+ break
+ if base_model:
+ break
+ if base_model:
+ base_model_box_style = "position: absolute; top: 4px; left: 4px; background-color: rgba(0, 0, 0, 0.8); color: white; padding: 2px 6px; border-radius: 4px; font-size: 0.7rem; z-index: 1;"
+ base_model_box = f"
{html.escape(base_model)}
"
+
onclick = f'cardClicked({item.get("prompt", None)})'
args = {
# "tabname": tabname,
@@ -344,6 +361,7 @@ class ExtraNetworksPage:
"version": item.get("version", ''),
"color": random_bright_color(),
"reference": "reference" if 'Reference' in item.get('name', '') else "",
+ "base_model_box": base_model_box,
}
# alias = item.get("alias", None)
# if alias is not None: