diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py
index 432dd41b3..911e22aae 100644
--- a/modules/ui_extra_networks.py
+++ b/modules/ui_extra_networks.py
@@ -25,6 +25,7 @@ debug = shared.log.trace if os.environ.get('SD_EN_DEBUG', None) is not None else
debug('Trace: EN')
card_full = '''
+ {base_model_box}
@@ -39,11 +40,11 @@ card_full = '''
'''
card_list = '''
@@ -324,6 +325,7 @@ class ExtraNetworksPage:
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):
@@ -334,6 +336,9 @@ class ExtraNetworksPage:
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 = {
@@ -354,9 +359,10 @@ class ExtraNetworksPage:
"card_click": item.get("onclick", '"' + html.escape(onclick) + '"'),
"mtime": item.get("mtime", 0),
"size": item.get("size", 0),
- "version": html.escape(base_model),
+ "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: